oweller2
commited on
Commit
•
eedfeda
1
Parent(s):
c116d9c
test
Browse files- modeling_flexbert.py +1 -0
modeling_flexbert.py
CHANGED
|
@@ -1506,6 +1506,7 @@ class FlexBertForMultipleChoice(FlexBertPreTrainedModel):
|
|
| 1506 |
|
| 1507 |
|
| 1508 |
class FlexBertForCasualLM(FlexBertPreTrainedModel):
|
|
|
|
| 1509 |
"""Bert Model transformer with a LM head.
|
| 1510 |
|
| 1511 |
This head is just a standard LM head module. Used for causal language modeling tasks.
|
|
|
|
| 1506 |
|
| 1507 |
|
| 1508 |
class FlexBertForCasualLM(FlexBertPreTrainedModel):
|
| 1509 |
+
config_class = FlexBertConfig # Add this line
|
| 1510 |
"""Bert Model transformer with a LM head.
|
| 1511 |
|
| 1512 |
This head is just a standard LM head module. Used for causal language modeling tasks.
|