From 38dd9d1555c93d727ff91e986ef48fd0689b7ba7 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 21 Mar 2025 17:00:35 -0700 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 5ad41aadcc..057d10d85e 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -654,7 +654,7 @@ def LlamaModel_fast_forward( inputs_embeds = self.embed_tokens(input_ids) inputs_embeds = inputs_embeds.to(_get_dtype(self.config.torch_dtype)) - print(inputs_embeds.shape) + print(inputs_embeds.shape, input_ids.shape, input_ids) # Normalized from Gemma IS_GEMMA = self.config.model_type.startswith("gemma")