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 18:00:51 +11:00
commit 5d88ffefd5
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

@ -310,7 +310,7 @@ def unsloth_save_model(
model.save_pretrained(**save_pretrained_settings)
# Update model tag
if push_to_hub:
if push_to_hub and hasattr(model, "config"):
_ = upload_to_huggingface(
model, save_pretrained_settings["save_directory"], token,
"finetuned", "trl", file_location = None,
@ -484,7 +484,7 @@ def unsloth_save_model(
print("Done.")
# Update model tag
if push_to_hub:
if push_to_hub and hasattr(model, "config"):
_ = upload_to_huggingface(
model, save_pretrained_settings["save_directory"], token,
"finetuned", "trl", file_location = None,