transformers==5.15.0
accelerate
torchvision
timm
einops
sentencepiece
pillow