westlake-repl
/

SaProt_650M_AF2

Model card Files Files and versions

LTEnjoy commited on Sep 2, 2024

Commit

2b951c3

·

verified ·

1 Parent(s): db47726

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ model = EsmForMaskedLM.from_pretrained(model_path)
 device = "cuda"
 model.to(device)
-seq = "MdEvVpQpLrVyQdYaKv"
 tokens = tokenizer.tokenize(seq)
 print(tokens)
@@ -28,7 +28,7 @@ outputs = model(**inputs)
 print(outputs.logits.shape)
 """
-['Md', 'Ev', 'Vp', 'Qp', 'Lr', 'Vy', 'Qd', 'Ya', 'Kv']
 torch.Size([1, 11, 446])
 """
 ```

 device = "cuda"
 model.to(device)
+seq = "M#EvVpQpL#VyQdYaKv" # Here "#" represents lower plDDT regions (plddt < 70)
 tokens = tokenizer.tokenize(seq)
 print(tokens)
 print(outputs.logits.shape)
 """
+['M#', 'Ev', 'Vp', 'Qp', 'L#', 'Vy', 'Qd', 'Ya', 'Kv']
 torch.Size([1, 11, 446])
 """
 ```