rm deprecated assertion

automodel.py

@@ -17,7 +17,7 @@ logger = logging.getLogger(__name__)
 
 class ClinicalMosaicForEmbeddingGeneration(BertPreTrainedModel):
 
-    def __init__(self, config, add_pooling_layer=False, **kwargs):
+    def __init__(self, config, **kwargs):
         """
         Initializes the BertEmbeddings class.
 
@@ -26,11 +26,8 @@ class ClinicalMosaicForEmbeddingGeneration(BertPreTrainedModel):
             add_pooling_layer (bool, optional): Whether to add a pooling layer. Defaults to False.
         """
         super().__init__(config)
-        assert (
-            config.num_hidden_layers >= config.num_embedding_layers
-        ), "num_hidden_layers should be greater than or equal to num_embedding_layers"
         self.config = config
-        self.bert = BertModel(config, add_pooling_layer=add_pooling_layer)
+        self.bert = BertModel(config, add_pooling_layer=False)
         # this resets the weights
         self.post_init()