From 1c7ecc83a70d070a73b2c7f9e117eba8b1de3abc Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 5 Mar 2025 03:38:16 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index f9b96fae4d..8ba7c45368 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -2644,7 +2644,7 @@ class FastLlamaModel: @staticmethod def for_training(model, use_gradient_checkpointing = True): if not hasattr(model, "parameters"): - raise TypeError("Unsloth: I think you're passing a tokenizer, not the model to for_inference!") + raise TypeError("Unsloth: I think you're passing a tokenizer, not the model to for_training!") # Delete all fast inference loras for param in model.parameters():