From bf3e10b26eece02d1aad11e2f8a83f019d1bfa92 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Wed, 21 Feb 2024 00:48:18 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/save.py b/unsloth/save.py index be105c3196..f625bd1890 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -234,7 +234,7 @@ def unsloth_save_model( old_username = None, ) - model.push_to_hub( + model.original_push_to_hub( repo_id = save_directory, use_temp_dir = use_temp_dir, commit_message = commit_message, @@ -248,7 +248,7 @@ def unsloth_save_model( tags = tags, ) if tokenizer is not None: - tokenizer.push_to_hub( + tokenizer.original_push_to_hub( repo_id = save_directory, use_temp_dir = use_temp_dir, commit_message = commit_message,