westlake-repl
/

ProTrek_650M

Model card Files Files and versions

LTEnjoy commited on May 22, 2024

Commit

b0bbd71

·

verified ·

1 Parent(s): 911d039

Update README.md

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -25,12 +25,12 @@ from utils.foldseek_util import get_struc_seq
 # Load model
 config = {
-    "protein_config": "weights/ProTrek_35M_UniRef50/esm2_t33_650M_UR50D",
-    "text_config": "weights/ProTrek_35M_UniRef50/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext",
-    "structure_config": "weights/ProTrek_35M_UniRef50/foldseek_t30_150M",
     "load_protein_pretrained": False,
     "load_text_pretrained": False,
-    "from_checkpoint": "weights/ProTrek_35M_UniRef50/ProTrek_650M_UniRef50.pt"
 }
 device = "cuda"

 # Load model
 config = {
+    "protein_config": "weights/ProTrek_650M_UniRef50/esm2_t33_650M_UR50D",
+    "text_config": "weights/ProTrek_650M_UniRef50/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext",
+    "structure_config": "weights/ProTrek_650M_UniRef50/foldseek_t30_150M",
     "load_protein_pretrained": False,
     "load_text_pretrained": False,
+    "from_checkpoint": "weights/ProTrek_650M_UniRef50/ProTrek_650M_UniRef50.pt"
 }
 device = "cuda"