torch>=2.1
torchvision>=0.16
pillow
huggingface_hub>=0.20
pyyaml
timm
