From 3d789cb09ee9f3e600c7bae1da040a930762d626 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 18 Feb 2024 22:59:37 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/unsloth/save.py b/unsloth/save.py index bd30c3020e..6f5f75cbd5 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -1048,10 +1048,14 @@ def patch_saving_functions(model): original_model.push_to_hub = types.MethodType(original_model.original_push_to_hub, original_model) pass original_model.original_push_to_hub = types.MethodType(original_model.push_to_hub, original_model) + if hasattr(original_model, "add_model_tags"): + original_model.add_model_tags(["unsloth",]) + pass if hasattr(original_model, "model"): original_model = original_model.model else: break pass + return model # And now re add our saving methods! original_push_to_hub = model.original_push_to_hub