torch>=2.1
einops>=0.7
numpy>=1.24
huggingface_hub>=0.23

[demo]
gradio>=5.29

[train]
pytorch-lightning>=2.1
gdown>=5.0
