From 74b69775c22b39baef13241fb3f2b52c9c757e86 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Mon, 29 Jan 2024 00:51:57 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/save.py b/unsloth/save.py index f5a0a31381..7190e10db6 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -1043,5 +1043,5 @@ def patch_saving_functions(model): model.push_to_hub_gguf = types.MethodType(unsloth_push_to_hub_gguf, model) model.save_pretrained_gguf = types.MethodType(unsloth_save_pretrained_gguf, model) pass - return + return model pass