Update modeling_protst.py

modeling_protst.py

@@ -235,7 +235,7 @@ class ProtSTForProteinPropertyPrediction(ProtSTPreTrainedModel):
         self.protein_model = EsmForProteinRepresentation(config.protein_config)
         self.text_model = BertForPubMed(config.text_config)
         self.logit_scale = nn.Parameter(torch.ones([]) * math.log(1 / 0.07))
-        self.classifier = ProtSTHead(config, out_dim=config.num_labels)
+        self.classifier = ProtSTHead(config.protein_config, out_dim=config.num_labels)
 
         self.post_init() # NOTE