From 89d2418cd785d2fbae2f07fb1ec00554e4022a9f Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 18 Feb 2024 23:35:18 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unsloth/save.py b/unsloth/save.py index 4ed797ea1f..2af1d63785 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -1045,9 +1045,9 @@ def patch_saving_functions(model): original_model = model while True: if hasattr(original_model, "original_push_to_hub"): - original_model.push_to_hub = original_model.original_push_to_hub, original_model + original_model.push_to_hub = original_model.original_push_to_hub pass - original_model.original_push_to_hub = original_model.push_to_hub, original_model + original_model.original_push_to_hub = original_model.push_to_hub if hasattr(original_model, "model"): original_model = original_model.model else: break @@ -1105,7 +1105,7 @@ def patch_saving_functions(model): while True: if not hasattr(original_model, "original_push_to_hub"): - original_model.original_push_to_hub = original_model.push_to_hub, original_model + original_model.original_push_to_hub = original_model.push_to_hub pass original_model.push_to_hub = types.MethodType(unsloth_push_to_hub, original_model)