torch
torchvision
diffusers