oweller2
commited on
Commit
•
f83d8e4
1
Parent(s):
8686e3f
update
Browse files- modeling_flexbert.py +0 -1
modeling_flexbert.py
CHANGED
|
@@ -1702,7 +1702,6 @@ class FlexBertForCasualLM(FlexBertPreTrainedModel):
|
|
| 1702 |
)
|
| 1703 |
|
| 1704 |
if self.pad_logits:
|
| 1705 |
-
# Reshape logits to 3D if needed
|
| 1706 |
new_logits = self.pad_inputs(logits, indices, batch_size, seq_len)[0]
|
| 1707 |
if len(new_logits.shape) == 2:
|
| 1708 |
new_logits = new_logits.unsqueeze(0)
|
|
|
|
| 1702 |
)
|
| 1703 |
|
| 1704 |
if self.pad_logits:
|
|
|
|
| 1705 |
new_logits = self.pad_inputs(logits, indices, batch_size, seq_len)[0]
|
| 1706 |
if len(new_logits.shape) == 2:
|
| 1707 |
new_logits = new_logits.unsqueeze(0)
|