ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update save.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-21 17:32:37 +11:00
commit da7b4f59ee
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/save.py
View file

@ -94,8 +94,9 @@ def fast_save_pickle(shard, name):
torch.save(
shard,
name,
pickle_module = pickle,
pickle_protocol = pickle.HIGHEST_PROTOCOL,
# HIGHEST_PROTOCOL seems to not work with Pytorch!
# pickle_module = pickle,
# pickle_protocol = pickle.HIGHEST_PROTOCOL,
)
return
pass