torch>=1.7
torchvision
pyyaml
huggingface_hub