torch>=2.6
torchvision>=0.21
numpy<2
Pillow
