einops>=0.3
torch>=1.6
gdown>=4.2.0
