torch>=2.1
safetensors>=0.4.2
omegaconf>=2.3.0
numpy>=1.26
einops>=0.7
huggingface-hub>=0.21