From f039eee50fe1a724413eb214094bcb3387579b33 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Tue, 19 Mar 2024 02:18:38 +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 01889517e4..707162dc55 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -540,6 +540,7 @@ def unsloth_save_model( # Save! save_location = save_pretrained_settings["save_directory"] + print("USERNAME = ", username, "SAVE_LOCATION = ", save_location) if username != save_location: save_pretrained_settings["save_directory"] = f"{username}/{save_location.lstrip('/')}" pass