mila-intel
/

ProtST-esm1b

Feature Extraction

Model card Files Files and versions

Jiqing commited on Jan 22, 2024

Commit

39d7a38

·

verified ·

1 Parent(s): fdc35b1

Update modeling_protst.py

Files changed (1) hide show

modeling_protst.py +0 -1

modeling_protst.py CHANGED Viewed

@@ -233,7 +233,6 @@ class ProtSTForProteinPropertyPrediction(ProtSTPreTrainedModel):
         self.config = config
         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.protein_config, out_dim=config.num_labels)

         self.config = config
         self.protein_model = EsmForProteinRepresentation(config.protein_config)
         self.logit_scale = nn.Parameter(torch.ones([]) * math.log(1 / 0.07))
         self.classifier = ProtSTHead(config.protein_config, out_dim=config.num_labels)