Update save.py

This commit is contained in:
Daniel Han-Chen 2024-03-19 19:26:15 +11:00
commit 74e98551d4

View file

@ -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 :(