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:35:11 +11:00
commit 83d906a2c0
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/save.py
View file

@ -841,7 +841,6 @@ def upload_to_huggingface(model, save_directory, token, method, extra = "", file
from huggingface_hub import whoami
try:
username = whoami(token = token)["name"]
print("Username = ", username, save_directory)
if type(old_username) is str and username != old_username:
username = old_username
pass
@ -851,7 +850,6 @@ def upload_to_huggingface(model, save_directory, token, method, extra = "", file
else:
username = save_directory.split("/")[0]
pass
print("Username = ", username, save_directory)
from huggingface_hub import create_repo
create_repo(
@ -1178,8 +1176,8 @@ def patch_saving_functions(model):
"finetuned", "trl", file_location = None,
old_username = None,
)
print("Saved to https://huggingface.co/" + arguments["repo_id"])
pass
print("Saved to https://huggingface.co/" + arguments["repo_id"])
return out
'''
exec(push_to_hub_text, globals())