Elgato_dark/unsloth
1
0
Fork 0
mirror of https://github.com/unslothai/unsloth synced 2026-04-21 13:37:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Update save.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-21 05:00:37 +11:00
parent 705cac0357
commit a6f4fb0075
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
unsloth/save.py
View file

@ -674,6 +674,7 @@ This {model_type} model was trained 2x faster with [Unsloth](https://github.com/
def upload_to_huggingface(model, save_directory, token, method, extra = "", file_location = None):
# Check for username
username = ""
save_directory = save_directory.lstrip("./")
if "/" not in save_directory:
from huggingface_hub import whoami
try:
@ -797,10 +798,10 @@ def unsloth_save_pretrained_gguf(
if push_to_hub:
print("Unsloth: Uploading GGUF to Huggingface Hub...")
username = upload_to_huggingface(
self, new_save_directory, token,
self, save_directory, token,
"GGUF converted", "gguf", file_location,
)
print(f"Saved to https://huggingface.co/{username}/{new_save_directory.lstrip('/')}")
print(f"Saved to https://huggingface.co/{username}/{new_save_directory.lstrip('/.')}")
pass
pass
@ -876,7 +877,7 @@ def unsloth_push_to_hub_gguf(
print("Unsloth: Uploading GGUF to Huggingface Hub...")
username = upload_to_huggingface(
self, new_save_directory, token,
self, repo_id, token,
"GGUF converted", "gguf", file_location,
)
print(f"Saved to https://huggingface.co/{username}/{new_save_directory.lstrip('/')}")