mila-intel
/

protst-esm1b-for-sequential-classification

Feature Extraction

Model card Files Files and versions Community

Jiqing commited on Jan 22

Commit

a1cddb1

•

1 Parent(s): cce60b4

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -7,7 +7,8 @@
     "AutoConfig": "Jiqing/ProtST-esm1b--configuration_protst.ProtSTConfig",
     "AutoModel": "Jiqing/ProtST-esm1b--modeling_protst.ProtSTForProteinPropertyPrediction"
   },
-  "model_type": "protst",
   "protein_config": {
     "_name_or_path": "/tmp/facebook/esm1b_t33_650M_UR50S",
     "architectures": [

     "AutoConfig": "Jiqing/ProtST-esm1b--configuration_protst.ProtSTConfig",
     "AutoModel": "Jiqing/ProtST-esm1b--modeling_protst.ProtSTForProteinPropertyPrediction"
   },
+  "model_type": "protest",
+  "num_labels": 2,
   "protein_config": {
     "_name_or_path": "/tmp/facebook/esm1b_t33_650M_UR50S",
     "architectures": [