gradio
torch==2.1.2
torchvision==0.16.2
triton==2.1.0
diffusers
transformers
accelerate