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-21 00:53:17 +11:00
commit 917f791ab7
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/save.py
View file

@ -1191,16 +1191,15 @@ def patch_saving_functions(model):
pass
try:
out = self.original_push_to_hub(**arguments)
self.original_push_to_hub(**arguments)
except:
del arguments["tags"]
out = self.original_push_to_hub(**arguments)
self.original_push_to_hub(**arguments)
pass
if hasattr(self, "config"):
print("Saved model to https://huggingface.co/" + arguments["repo_id"])
pass
return out
'''
exec(push_to_hub_text, globals())