accelerate
f8c77f05 - Revert default behavior of `get_state_dict_from_offload` (#3253)

Commit
1 year ago
Revert default behavior of `get_state_dict_from_offload` (#3253) * change default to None Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> * introduce move_to_device argument Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> * remove move_to_device Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> --------- Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
Author
Parents
Loading