transformers
diffusers
torch
torchvision
accelerate