oweller2
commited on
Commit
•
1866ba4
1
Parent(s):
ce41753
update
Browse files- modeling_flexbert.py +1 -0
modeling_flexbert.py
CHANGED
|
@@ -1505,6 +1505,7 @@ class FlexBertForMultipleChoice(FlexBertPreTrainedModel):
|
|
| 1505 |
|
| 1506 |
|
| 1507 |
class FlexBertForCausalLM(FlexBertPreTrainedModel):
|
|
|
|
| 1508 |
"""Bert Model transformer with a LM head.
|
| 1509 |
|
| 1510 |
This head is just a standard LM head module. Used for causal language modeling tasks.
|
|
|
|
| 1505 |
|
| 1506 |
|
| 1507 |
class FlexBertForCausalLM(FlexBertPreTrainedModel):
|
| 1508 |
+
config_class = FlexBertConfig
|
| 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.
|