add config class

automodel.py

@@ -11,11 +11,13 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 
 from .bert_layers_mosa import BertModel
+from .configuration_bert import BertConfig
 
 logger = logging.getLogger(__name__)
 
 
 class ClinicalMosaicForEmbeddingGeneration(BertPreTrainedModel):
+    config_class = BertConfig
 
     def __init__(self, config, **kwargs):
         """
@@ -95,6 +97,7 @@ class ClinicalMosaicForSequenceClassification(BertPreTrainedModel):
     """Bert Model transformer with a sequence classification/regression head.
     This head is just a linear layer on top of the pooled output.
     """
+    config_class = BertConfig
 
     def __init__(self, config, **kwargs):
         super().__init__(config)