search

evo-design / evo

Biological foundation modeling from molecular to genome scale
Apache License 2.0
874 stars 99 forks source link

TORCH_HOME not used when downloading models #48

Closed james-vincent closed 2 months ago

james-vincent commented 3 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 2 months ago

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

Zymrael commented 2 months ago

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