transformers
gradio
torch
Pillow
einops
torchvision
accelerate>=0.26.0