diff --git a/unsloth/save.py b/unsloth/save.py index 0a48866b3e..64342d91de 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -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