torch>=2.0.0
torchvision
transformers>=4.40.0
pillow
numpy