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 2024-11-21 03:47:57 -08:00
commit e0b14fab34
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

@ -2077,7 +2077,7 @@ def unsloth_generic_save(
maximum_memory_usage : float = 0.9,
):
if token is None and push_to_hub: token = get_token()
merge_and_overwrite_lora(
get_model_name,
create_huggingface_repo,
@ -2085,7 +2085,7 @@ def unsloth_generic_save(
save_location = save_directory,
push_to_hub = push_to_hub,
token = token,
upload_location = save_directory,
upload_location = save_directory if push_to_hub else None,
low_disk_space_usage = True,
private = private,
)