transformers 
torch 
Pillow
torchvision 
einops
accelerate