facebookresearch / esm

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

Update torch-hub load to point to main branch #138

Closed liujas000 closed 3 years ago

liujas000 commented 3 years ago

Now that the primary branch is called main, torch.hub calls must explicitly reference the main branch

In the README, this should be

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

Reference: https://stackoverflow.com/questions/69349308/torch-hub-load-raises-httperror-http-error-404-not-found-when-loading-model