torch>=2.5.0
transformers
einops
pycocotools
pillow
