Update save.py
This commit is contained in:
parent
def91f0195
commit
74e98551d4
1 changed files with 3 additions and 0 deletions
|
|
@ -574,6 +574,9 @@ 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 :(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue