diffusers
torch
torchvision
accelerate