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-19 03:39:09 +11:00
commit aa45208f5a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/save.py
View file

@ -219,7 +219,7 @@ def unsloth_save_model(
else:
commit_description += " (Trained with Unsloth 2x faster)"
pass
model.push_to_hub(
repo_id = save_directory,
use_temp_dir = use_temp_dir,
@ -492,7 +492,7 @@ def unsloth_save_model(
for _ in range(3):
torch.cuda.empty_cache()
gc.collect()
return save_directory
return save_directory, username
pass