Update modeling_bert.py

modeling_bert.py

@@ -68,6 +68,8 @@ class BertClsPooler(nn.Module):
 
 class BertModel(BertPreTrainedModel):
 
+    config_class = BertClsConfig
+
     def __init__(self, config, add_pooling_layer=True):
         super().__init__(config)
         self.config = config
@@ -199,6 +201,8 @@ class BertModel(BertPreTrainedModel):
 
 class BertForSequenceClassification(BertPreTrainedModel):
 
+    config_class = BertClsConfig
+
     def __init__(self, config):
         super().__init__(config)
         self.num_labels = config.num_labels