diffusers
torch
torchvision