diffusers
gradio
torch  # or torch torchvision torchaudio if you're using GPU
transformers