Update modeling_phi3.py

modeling_phi3.py

@@ -1302,8 +1302,8 @@ class Phi3ForCausalLM(Phi3PreTrainedModel):
         loss = None
         if labels is not None:
             print("===test===")
-            print(labels.view(-1).shape)
-            print(labels.view(-1)[labels.view(-1) == -100].shape)
+            print(labels[0].shape)
+            print(labels[0])
             # Shift so that tokens < n predict n
             shift_logits = logits[..., :-1, :].contiguous()
             shift_labels = labels[..., 1:].contiguous()