Fix AutoModel not loading model correctly due to config_class inconsistency

dnabert_layer.py

@@ -9,22 +9,38 @@ from transformers.models.bert.modeling_bert import BertForPreTraining as Transfo
 from transformers.models.bert.modeling_bert import BertPreTrainedModel
 from transformers.modeling_outputs import SequenceClassifierOutput
 
+from .configuration_bert import BertConfig
+
 
 class BertModel(TransformersBertModel):
+    
+    config_class = BertConfig
+    
     def __init__(self, config):
         super().__init__(config)
+        
 
 class BertForMaskedLM(TransformersBertForMaskedLM):
+    
+    config_class = BertConfig
+    
     def __init__(self, config):
         super().__init__(config)
+        
 
 class BertForPreTraining(TransformersBertForPreTraining):
+    
+    config_class = BertConfig
+    
     def __init__(self, config):
         super().__init__(config)
 
 
 
 class DNABertForSequenceClassification(BertPreTrainedModel):
+    
+    config_class = BertConfig
+    
     def __init__(self, config):
         super().__init__(config)
         self.num_labels = config.num_labels