Upload BertForMaskedLM
Browse files- modeling_bert.py +1 -1
modeling_bert.py
CHANGED
|
@@ -852,7 +852,7 @@ class BertLMHeadModel(BertPreTrainedModel):
|
|
| 852 |
|
| 853 |
class BertForMaskedLM(BertPreTrainedModel):
|
| 854 |
|
| 855 |
-
|
| 856 |
|
| 857 |
def __init__(self, config):
|
| 858 |
super().__init__(config)
|
|
|
|
| 852 |
|
| 853 |
class BertForMaskedLM(BertPreTrainedModel):
|
| 854 |
|
| 855 |
+
config_class = BertConfig
|
| 856 |
|
| 857 |
def __init__(self, config):
|
| 858 |
super().__init__(config)
|