pub_soft/esm
3
0
Fork 0
mirror of https://github.com/evolutionaryscale/esm.git synced 2026-06-04 17:14:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix examples/raw_forward.py

Browse Source
This commit is contained in:
Zeming Lin
2024-08-06 16:17:44 +00:00
parent 8c0e910158
commit 10077d8a8e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/raw_forwards.py
View File

@@ -26,7 +26,7 @@ def inverse_folding_example():
encoder = ESM3_structure_encoder_v0("cuda")
model = ESM3_sm_open_v0("cuda")
chain = ProteinChain.from_pdb("esm/data/1utn.pdb")
chain = ProteinChain.from_rcsb("1utn", "A")
coords, plddt, residue_index = chain.to_structure_encoder_inputs()
coords = coords.cuda()
plddt = plddt.cuda()