diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 852388e4f8..9ce868150f 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -635,6 +635,7 @@ def LlamaForCausalLM_fast_forward( logits = self.lm_head(hidden_states) loss = None + print(labels) if labels is not None: shift_logits = logits if not hasattr(self, "extra_ignored_labels"):