From b1b31c93ae38bd2b8f80e7c8c84784211a6c28aa Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Tue, 19 Mar 2024 03:36:12 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/save.py b/unsloth/save.py index d12235f473..0dfd9ce87f 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -579,6 +579,7 @@ def unsloth_save_model( # Sadly .save_pretrained doesn't work :( # We first save it via .save_pretrained, then upload manually! save_pretrained_settings["save_directory"] = new_save_directory + save_pretrained_settings["push_to_hub"] = False internal_model.save_pretrained(**save_pretrained_settings) # Now manually go through each file and upload them manually!