Update modeling_phi3.py
Browse files- 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
|