torch>=1.7
torchvision
pyyaml
huggingface_hub>=0.17.0
safetensors>=0.2
numpy
