Satandon1999 commited on
Commit
9e12a92
1 Parent(s): 89a52d1

Update modeling_phi3.py

Browse files
Files changed (1) hide show
  1. modeling_phi3.py +1 -0
modeling_phi3.py CHANGED
@@ -1302,6 +1302,7 @@ class Phi3ForCausalLM(Phi3PreTrainedModel):
1302
  loss = None
1303
  if labels is not None:
1304
  print("===test===")
 
1305
  print(labels[0].shape)
1306
  print(labels[0])
1307
  # Shift so that tokens < n predict n
 
1302
  loss = None
1303
  if labels is not None:
1304
  print("===test===")
1305
+ torch.set_printoptions(profile="full")
1306
  print(labels[0].shape)
1307
  print(labels[0])
1308
  # Shift so that tokens < n predict n