oweller2
commited on
Commit
•
855df5e
1
Parent(s):
37eee20
updates
Browse files- modeling_flexbert.py +1 -0
modeling_flexbert.py
CHANGED
|
@@ -1701,6 +1701,7 @@ class FlexBertForCasualLM(FlexBertPreTrainedModel):
|
|
| 1701 |
shift_labels.view(-1)
|
| 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:
|
|
|
|
| 1701 |
shift_labels.view(-1)
|
| 1702 |
)
|
| 1703 |
|
| 1704 |
+
assert False
|
| 1705 |
if self.pad_logits:
|
| 1706 |
new_logits = self.pad_inputs(logits, indices, batch_size, seq_len)[0]
|
| 1707 |
if len(new_logits.shape) == 2:
|