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,