Satandon1999
commited on
Commit
•
9e12a92
1
Parent(s):
89a52d1
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
|