Update automodel.py

automodel.py

@@ -84,7 +84,7 @@ class MosaicBertForEmbeddingGeneration(BertPreTrainedModel):
         # batch_size, hidden_dim
         return encoder_outputs_all
 
-class MosaicBertForSequenceClassification(BertPreTrainedModel):
+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.
     """