gradio
torchvision
diffusers
einops
torch