torch
torchvision
diffusers
