Add move_to_device
kwarg to the optimizer's load_state_dict
#1344
Merged
matthewdouglas merged 1 commit intobitsandbytes-foundation:mainfrom koute:mainSep 19, 2024
+5-2
move_to_device
kwarg to the optimizer's load_state_dict
#1344