Update automodel.py

automodel.py

@@ -103,7 +103,7 @@ class ClinicalMosaicForSequenceClassification(BertPreTrainedModel):
         super().__init__(config)
         self.num_labels = config.num_labels
         self.config = config
-        self.bert = BertModel(config, add_pooling_layer=False)
+        self.bert = BertModel(config, add_pooling_layer=True)
         classifier_dropout = (
             config.classifier_dropout
             if config.classifier_dropout is not None