From 7ee8243e8bc542087891c39b83d1fc0c82c4fec3 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 18 Feb 2024 20:19:28 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/unsloth/save.py b/unsloth/save.py index 4d09a97c8a..601aadea6a 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -1059,7 +1059,9 @@ def patch_saving_functions(model): pass # And now re add our saving methods! + print(1) original_push_to_hub = model.original_push_to_hub + print(2) signature = str(inspect.signature(original_push_to_hub)).replace("NoneType", "None") signature = signature[1:] signature = re.sub("", "torch.save", signature)