transformers
diffusers
gradio
torch
torchvision
torchtext
accelerate
numpy