evo-design / evo

Biological foundation modeling from molecular to genome scale

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TORCH_HOME not used when downloading models

james-vincent opened this issue · comments

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.

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

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