ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update save.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-02-20 02:19:36 +11:00
commit 3728b8e876
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/save.py
View file

@ -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: