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-21 00:48:18 +11:00
commit bf3e10b26e
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

@ -234,7 +234,7 @@ def unsloth_save_model(
old_username = None,
)
model.push_to_hub(
model.original_push_to_hub(
repo_id = save_directory,
use_temp_dir = use_temp_dir,
commit_message = commit_message,
@ -248,7 +248,7 @@ def unsloth_save_model(
tags = tags,
)
if tokenizer is not None:
tokenizer.push_to_hub(
tokenizer.original_push_to_hub(
repo_id = save_directory,
use_temp_dir = use_temp_dir,
commit_message = commit_message,