facebookresearch / esm

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

ESM2 pretrained model in torch hub not accessible #525

Closed lordim closed 1 year ago

lordim commented 1 year ago

I tried downloading and using ESM2 pretrained model using the code on README:

import torch
model, alphabet = torch.hub.load("facebookresearch/esm:main", "esm2_t33_650M_UR50D")

However, it is throwing me the following error: RuntimeError: Cannot find callable esm2_t33_650M_UR50D in hubconf

It seems that the code works for an older version ESM (ESM-1b), but not so for ESM2.