search

evolutionaryscale / esm

Other
1.17k stars 128 forks source link

Importing ESM2 and ESM3 #12

Open yashaektefaie opened 3 months ago

yashaektefaie commented 3 months ago

For benchmarking purposes I would like to import both ESM2 and ESM3 but the problem is both have the same import name "esm" so I am unable to import ESM2. How should I go about importing both? "esm" does not encompass ESM2 and ESM3, both are still separate libraries right?

tb1over commented 3 months ago

use different virtual env ?

yashaektefaie commented 3 months ago

Yes beyond that I was wondering if there was a way of doing it without having to have separate ESM2 and ESM3 virtual environments? Specifically if there was plans to be able to import ESM2 models since they share the same import name

ebetica commented 3 months ago

🙈 I can't think of a good way around this. We'll work on a solution. But in the mean time, you might just want to run something in root of this repo like

sed -i "s/esm./esm3./g" **/*.py
sed -i "s/esm/esm3/g" pyproject.toml
mv esm esm3

and then pip install . as a workaround?

JonasLi-19 commented 3 months ago

hello developer, I want to know if I can use esm3's weight file in a esm2-based model. My teacher asked me to load esm3's weight in a previous model which uses pretrained esm2. I don't think it is plausible as they might have different model structures, right?

ebetica commented 2 months ago

No, you cannot use esm3 weight file in a esm2-based model.