From 83d906a2c0ec54c072c18dcf539803713b84704e Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Tue, 20 Feb 2024 18:35:11 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/unsloth/save.py b/unsloth/save.py index 221ccb3e76..7fb8d5c55d 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -841,7 +841,6 @@ def upload_to_huggingface(model, save_directory, token, method, extra = "", file from huggingface_hub import whoami try: username = whoami(token = token)["name"] - print("Username = ", username, save_directory) if type(old_username) is str and username != old_username: username = old_username pass @@ -851,7 +850,6 @@ def upload_to_huggingface(model, save_directory, token, method, extra = "", file else: username = save_directory.split("/")[0] pass - print("Username = ", username, save_directory) from huggingface_hub import create_repo create_repo( @@ -1178,8 +1176,8 @@ def patch_saving_functions(model): "finetuned", "trl", file_location = None, old_username = None, ) + print("Saved to https://huggingface.co/" + arguments["repo_id"]) pass - print("Saved to https://huggingface.co/" + arguments["repo_id"]) return out ''' exec(push_to_hub_text, globals())