torchvision
diffusers
imageio