fix labels

modeling_pharia.py

@@ -774,7 +774,7 @@ class PhariaForCausalLM(PhariaPreTrainedModel):
         if self.training and labels is not None:
             # Shift logits and labels for causal language modeling
             shift_logits = logits[..., :-1, :].contiguous()
-            shift_labels = outputs['labels'][..., 1:].contiguous()
+            shift_labels = labels[..., 1:].contiguous()
     
             # Flatten the tokens
             shift_logits = shift_logits.view(-1, shift_logits.size(-1))