Bug fix

unslothai · Dec 26, 2024 · 1d7411a · 1d7411a
1 parent ae090f3
commit 1d7411a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/unsloth_zoo/__init__.py b/unsloth_zoo/__init__.py
@@ -14,7 +14,7 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-__version__ = "2024.12.5"
+__version__ = "2024.12.6"
 
 from importlib.util import find_spec
 if find_spec("unsloth") is None:

diff --git a/unsloth_zoo/saving_utils.py b/unsloth_zoo/saving_utils.py
@@ -595,7 +595,7 @@ def upload_items(filename = None):
             os.remove(os.path.join(save_directory, filename)) # Remove to conserve disk space
         pass
     pass
-    if not low_disk_space_usage: upload_items()
+    if not low_disk_space_usage and push_to_hub: upload_items()
 
     # Check for errors
     if len(lora_weights) != n_saved_modules: