chaidiscovery / chai-lab

Chai-1, SOTA model for biomolecular structure prediction
https://www.chaidiscovery.com
Other
1.24k stars 156 forks source link

Chai-1 Embeddings #32

Open MelihDarcanxyz opened 2 months ago

MelihDarcanxyz commented 2 months ago

Hi, thanks for this awesome repo and paper. I have a question.

Is it possible to get Chai-1 embeddings of given sequences? From what I see, it's only possible to get .pdb outputs.

Thanks in advance.

arogozhnikov commented 2 months ago

Hi Melih,

we don't expose any embeddings from the model, but you can modify code to save or return these two tensors

https://github.com/chaidiscovery/chai-lab/blob/90da6fcd77b9520caeac5f3220cb3edb760c6a47/chai_lab/chai1.py#L434