cannot import name 'esm_if1_gvp4_t16_142M_UR50'

[fisevnen]

Hello,
When I started to use ESM2, some problems appeared.

Here is my code:
model, alphabet = esm.pretrained.esm2_t33_650M_UR50D()
And it returned:
AttributeError: module 'esm.pretrained' has no attribute 'esm2_t33_650M_UR50D'

Then I used torch hub to download, here is my code:
model, alphabet = torch.hub.load("facebookresearch/esm:main", "esm2_t33_650M_UR50D")
And it returned:
ImportError: cannot import name 'esm_if1_gvp4_t16_142M_UR50' from 'esm.pretrained' (d:\anaconda\envs\test_env\lib\site-packages\esm\pretrained.py)

Could you please help with this?

[fisevnen]

I got where the problem is.
My fair-esm version is too old. Once I upgraded it, problem solved.