mila-intel
/

protst-esm1b-for-sequential-classification

Feature Extraction

Model card Files Files and versions

Jiqing commited on Jan 22, 2024

Commit

09326a0

·

verified ·

1 Parent(s): f3e5b30

Update config.json

Files changed (1) hide show

config.json +3 -2

config.json CHANGED Viewed

@@ -1,11 +1,11 @@
 {
   "_name_or_path": "Jiqing/ProtST-esm1b",
   "architectures": [
-    "ProtSTModel"
   ],
   "auto_map": {
     "AutoConfig": "Jiqing/ProtST-esm1b--configuration_protst.ProtSTConfig",
-    "AutoModel": "Jiqing/ProtST-esm1b--modeling_protst.ProtSTModel"
   },
   "model_type": "protst",
   "protein_config": {
@@ -40,6 +40,7 @@
     "model_type": "bert",
     "sep_token_id": 3
   },
   "torch_dtype": "float32",
   "transformers_version": "4.36.2"
 }

 {
   "_name_or_path": "Jiqing/ProtST-esm1b",
   "architectures": [
+    "ProtSTForProteinPropertyPrediction"
   ],
   "auto_map": {
     "AutoConfig": "Jiqing/ProtST-esm1b--configuration_protst.ProtSTConfig",
+    "AutoModel": "Jiqing/ProtST-esm1b--modeling_protst.ProtSTForProteinPropertyPrediction"
   },
   "model_type": "protst",
   "protein_config": {
     "model_type": "bert",
     "sep_token_id": 3
   },
+  "num_labels": 2,
   "torch_dtype": "float32",
   "transformers_version": "4.36.2"
 }