torch>=2.0
torchvision
numpy
einops
huggingface_hub
safetensors
