Pass params from saved model

model.py

@@ -21,9 +21,9 @@ class BertMesh(PreTrainedModel):
         config,
         pretrained_model="microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract",
         num_labels=28761,
-        hidden_size=512,
+        hidden_size=1024,
         dropout=0,
-        multilabel_attention=False,
+        multilabel_attention=True,
     ):
         super().__init__(config=config)
         self.config.auto_map = {"AutoModel": "transformers_model.BertMesh"}