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 03:31:44 +11:00
commit 0d80335998
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
unsloth/save.py
View file

@ -528,13 +528,16 @@ def unsloth_save_model(
# First check if we're pushing to an organization!
save_directory = save_pretrained_settings["save_directory"]
new_save_directory, new_username = _determine_username(save_directory, username, token)
if token is not None:
from huggingface_hub import whoami
actual_username = whoami(token = token)["name"]
else:
actual_username = username
if save_pretrained_settings["push_to_hub"]:
new_save_directory, new_username = _determine_username(save_directory, username, token)
if token is not None:
from huggingface_hub import whoami
actual_username = whoami(token = token)["name"]
else:
actual_username = username
pass
# Check if pushing to an organization
if save_pretrained_settings["push_to_hub"] and (username != actual_username):