From 74e98551d4fd64e398981958d8fcd63f847df68d Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Tue, 19 Mar 2024 19:26:15 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/unsloth/save.py b/unsloth/save.py index 17116887e1..499fbb67a1 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -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 :(