From 53f6a07b92220cd0e8ee16b9ef32f5cc235d49da Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 18 Feb 2024 20:22:52 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/unsloth/save.py b/unsloth/save.py index 601aadea6a..3f179e289b 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -1046,7 +1046,11 @@ def patch_saving_functions(model): while True: if hasattr(original_model, "original_push_to_hub"): original_model.push_to_hub = types.MethodType(original_model.original_push_to_hub, original_model) - del original_model.original_push_to_hub + print(original_model) + try: + del original_model.original_push_to_hub + except: + print("!") if hasattr(original_model, "push_to_hub_merged"): del original_model.push_to_hub_merged if hasattr(original_model, "save_pretrained_merged"): del original_model.save_pretrained_merged if hasattr(original_model, "push_to_hub_gguf"): del original_model.push_to_hub_gguf