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 04:44:17 +11:00
commit 164fa807fe
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/save.py
View file

@ -256,6 +256,7 @@ def unsloth_save_model(
"finetuned", "trl", file_location = None,
old_username = None,
)
print("Saved to https://huggingface.co/" + save_directory)
return save_directory
pass
@ -315,6 +316,7 @@ def unsloth_save_model(
"finetuned", "trl", file_location = None,
old_username = None,
)
print("Saved to https://huggingface.co/" + save_pretrained_settings["save_directory"])
pass
print(" Done.")