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-18 22:03:22 +11:00
commit b6a6e90b7a
1 changed files with 18 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

32
unsloth/save.py
View file

@ -313,7 +313,7 @@ def unsloth_save_model(
)
save_pretrained_settings["save_directory"] = new_save_directory
tokenizer_save_settings ["save_directory"] = new_save_directory
# tokenizer_save_settings ["save_directory"] = new_save_directory
save_directory = new_save_directory
pass
@ -1074,21 +1074,23 @@ def patch_saving_functions(model):
elif hasattr(self, "add_model_tags"):
self.add_model_tags(["unsloth",])
commit_message = arguments["commit_message"]
commit_description = arguments["commit_description"]
if commit_message is not None:
if not commit_message.endswith(" "): commit_message += " "
commit_message += "(Trained with Unsloth)"
else:
commit_message = "Upload model trained with Unsloth"
arguments["commit_message"] = commit_message
if "commit_message" in arguments:
commit_message = arguments["commit_message"]
if commit_message is not None:
if not commit_message.endswith(" "): commit_message += " "
commit_message += "(Trained with Unsloth)"
else:
commit_message = "Upload model trained with Unsloth"
arguments["commit_message"] = commit_message
if commit_description is not None:
if not commit_description.endswith(" "): commit_description += " "
commit_description += "(Trained with Unsloth)"
else:
commit_description = "Upload model trained with Unsloth"
arguments["commit_description"] = commit_description
if "commit_description" in arguments:
commit_description = arguments["commit_description"]
if commit_description is not None:
if not commit_description.endswith(" "): commit_description += " "
commit_description += "(Trained with Unsloth)"
else:
commit_description = "Upload model trained with Unsloth"
arguments["commit_description"] = commit_description
try:
return self.original_push_to_hub(**arguments)