search

evo-design / evo

Biological foundation modeling from molecular to genome scale
Apache License 2.0
1.14k stars 137 forks source link

TORCH_HOME not used when downloading models #48

Closed james-vincent closed 7 months ago

james-vincent commented 8 months ago

Models are downloaded to $HOME/.cache by default. Is there a way to change this? Setting TORCH_HOME does not work.

We will deploy evo on a cluster. Having each user download the models is a waste of space.

adamcatto commented 7 months ago

I had the same problem – resolved it by setting the HF_HOME environment variable.

Zymrael commented 7 months ago

This is correct, the Evo class uses HF to download models, so use HF environment variables