gradio
torch
transformers
accelerate
spaces
# flash-attn
git+https://github.com/Disty0/diffusers
timm
torchvision
einops