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>