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-03-19 19:36:44 +11:00
commit fa5cd5fddc
1 changed files with 0 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/save.py
View file

@ -574,9 +574,6 @@ def unsloth_save_model(
# Check if pushing to an organization
if save_pretrained_settings["push_to_hub"] and (username != actual_username):
for key, value in save_pretrained_settings.items():
if key == "state_dict": continue
print(key, value)
print(f"Unsloth: Saving to organization with address {new_save_directory}")
# Pushing to organization!
# Sadly .save_pretrained doesn't work :(
@ -601,9 +598,6 @@ def unsloth_save_model(
ignore_patterns = "*.md",
)
else:
for key, value in save_pretrained_settings.items():
if key == "state_dict": continue
print(key, value)
internal_model.save_pretrained(**save_pretrained_settings)
pass