torch
torchvision
transformers>=4.40
safetensors
pillow
numpy
tqdm
einops
accelerate
sentencepiece