Update modeling_tunbert.py

modeling_tunbert.py

@@ -14,6 +14,7 @@ class classifier(nn.Module):
 
 
 class TunBERT(PreTrainedModel):
+    config_class = AutoConfig
     def __init__(self, config):
         super().__init__(config)
         self.BertModel = BertModel(config)