torch
gradio
accelerate
transformers==4.41.0
pillow
einops
torchvision
matplotlib
numpy