Update llama.py

2025-03-21 17:06:06 -07:00 · 2025-03-21 17:06:06 -07:00 · 0a9bfc59b7
commit 0a9bfc59b7
parent e99808d571
1 changed files with 1 additions and 0 deletions
--- 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,