Maxscha
/

bert-test-aligned

Model card Files Files and versions

Maxscha commited on Aug 16, 2024

Commit

b6ad527

·

verified ·

1 Parent(s): ba6466e

Upload BertForMaskedLM

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -852,7 +852,7 @@ class BertLMHeadModel(BertPreTrainedModel):
 class BertForMaskedLM(BertPreTrainedModel):
-    model_config = BertConfig
     def __init__(self, config):
         super().__init__(config)

 class BertForMaskedLM(BertPreTrainedModel):
+    config_class = BertConfig
     def __init__(self, config):
         super().__init__(config)