From da7b4f59ee0a4e2dcb8d5177f8603800fd281b2c Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Sun, 21 Jan 2024 17:32:37 +1100
Subject: [PATCH] Update save.py

---
 unsloth/save.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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