facebookresearch / esm

Evolutionary Scale Modeling (esm): Pretrained language models for proteins
MIT License
3.26k stars 643 forks source link

EMSIF1-Encoder output as structure representation #443

Open LDAIprotein opened 1 year ago

LDAIprotein commented 1 year ago

rep = esm.inverse_folding.multichain_util.get_encoder_output_for_complex( model, alphabet, coords, target_chain_id). I found that target_chain_id can only be a single chain. Can the input (target_chain_id) improve the code to get multi-chain structure embeddings?