transformers>=5.8
accelerate
torch
torchvision
Pillow
sentencepiece
einops
numpy