From 77e4034de6d20d21cae75c13071c07412297647e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 3 Nov 2024 18:08:22 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 4e83e69d60..3c4d8f3b38 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1804,6 +1804,7 @@ class FastLlamaModel: @staticmethod def post_patch(model, tokenizer): model, tokenizer = patch_model_and_tokenizer(model, tokenizer, downcast_rope = True) + return model, tokenizer pass