diff --git a/unsloth/save.py b/unsloth/save.py index 9e595fe7de..ae7b5252e3 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -448,7 +448,7 @@ def unsloth_save_model( # commit_description does not seem to work? what_to_delete = ("use_temp_dir", "commit_message", "create_pr", "revision", "commit_description", "tags",) \ - if username is None else \ + if not push_to_hub else \ ("use_temp_dir", "create_pr", "revision", "tags", "commit_description",) for deletion in what_to_delete: del save_pretrained_settings[deletion] @@ -1148,7 +1148,7 @@ def patch_saving_functions(model): if not commit_message.endswith(" "): commit_message += " " commit_message += "(Trained with Unsloth 2x faster)" else: - commit_message = "Upload model trained with Unsloth" + commit_message = "Upload model trained with Unsloth 2x faster" arguments["commit_message"] = commit_message if "commit_description" in arguments: