torch
transformers
numpy
Pillow
einops
timm==0.6.12