YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
ISM
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
Quickstart
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-300m-2024-12 --local-dir /path/to/save/ismc
# alternative
git clone https://huggingface.co/jozhang97/ismc-300m-2024-12
Add the following lines of code.
from esm.models.esmc import ESMC
model = ESMC.from_pretrained("esmc_300m").to("cuda")
state_dict = torch.load('/path/to/ismc_300m_2024_12_v0.pth')
model.load_state_dict(state_dict)