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-03-19 02:18:38 +11:00
commit f039eee50f
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
unsloth/save.py
View file

@ -540,6 +540,7 @@ def unsloth_save_model(
# Save!
save_location = save_pretrained_settings["save_directory"]
print("USERNAME = ", username, "SAVE_LOCATION = ", save_location)
if username != save_location:
save_pretrained_settings["save_directory"] = f"{username}/{save_location.lstrip('/')}"
pass