From 0a9bfc59b7f7322d2d4940b7548c947b9c937fbc Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 21 Mar 2025 17:06:06 -0700
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 8e7f25f589..d20e2b4b75 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -1201,6 +1201,7 @@ def PeftModelForCausalLM_fast_forward(
     logits_to_keep = 0,
     **kwargs,
 ):
+    print(input_ids, input_ids.shape)
     return self.base_model(
         input_ids = input_ids,
         causal_mask = causal_mask,