Update modeling_tunbert.py

modeling_tunbert.py

@@ -1,7 +1,7 @@
 import torch.nn as nn
 from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModelForSequenceClassification, PreTrainedModel,AutoConfig, BertModel
 from transformers.modeling_outputs import SequenceClassifierOutput
-
+from transformers.models.bert.configuration_bert import BertConfig
 class classifier(nn.Module):
   def __init__(self,config):
     super().__init__()
@@ -14,7 +14,7 @@ class classifier(nn.Module):
 
 
 class TunBERT(PreTrainedModel):
-    config_class = AutoConfig
+    config_class = BertConfig
     def __init__(self, config):
         super().__init__(config)
         self.BertModel = BertModel(config)