From c528d8c44a0891aa1a3b26d7b8e2bcc45bcd3117 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Feb 2025 00:26:58 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 0b567b023a..2d5e43ba64 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1189,7 +1189,7 @@ def PeftModelForCausalLM_fast_forward( num_logits_to_keep=0, **kwargs, ): - return self.base_model( + a = self.base_model( input_ids=input_ids, causal_mask=causal_mask, attention_mask=attention_mask, @@ -1201,6 +1201,7 @@ def PeftModelForCausalLM_fast_forward( num_logits_to_keep=num_logits_to_keep, **kwargs, ) + print(a) pass