From e99808d5717b241896bc578baa8dcdacd2c77e94 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 21 Mar 2025 17:05:43 -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 24f5ba2001..8e7f25f589 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1039,7 +1039,7 @@ def CausalLM_fast_forward(fast_forward_inference): output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states ) return_dict = return_dict if return_dict is not None else self.config.use_return_dict - + print(input_ids.shape, input_ids) # decoder outputs consists of (dec_features, layer_state, dec_hidden, dec_attn) self.model._has_no_labels = labels is None outputs = self.model(