Downloads · 30 days
0
jozhang97/ismc-600m-2024-12
ismc-600m-2024-12 is a machine learning model from jozhang97. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
By Jeffrey Ouyang-Zhang, Chengyue Gong, Yue Zhao, Philipp Krähenbühl, Adam Klivans, Daniel J. Diaz
Downloads · 30 days
0
Access
Public
Updated Dec 6, 2024
Repo size
2.3 GB
Likes
1
Public
Click a slice to open those files.
.pth2.3 GB · 100%
From the Hugging Face model README
By Jeffrey Ouyang-Zhang, Chengyue Gong, Yue Zhao, Philipp Krähenbühl, Adam Klivans, Daniel J. Diaz
This repository contains the model presented in the paper Distilling Structural Representations into Protein Sequence Models. The official github can be found at https://github.com/jozhang97/ism.
TL; DR. ESM2 with enriched structural representations
This quickstart assumes that the user is already working with ESM-C and is interested in replacing ESM-C with ISM-C. First, download ISM-C.
# recommended
huggingface-cli download jozhang97/ismc-600m-2024-12 --local-dir /path/to/save/ismc
# alternative
git clone https://huggingface.co/jozhang97/ismc-600m-2024-12
Add the following lines of code.
from esm.models.esmc import ESMC
model = ESMC.from_pretrained("esmc_600m").to("cuda")
state_dict = torch.load('/path/to/ismc_600m_2024_12_v0.pth')
model.load_state_dict(state_dict)