diffusers
pygame
torch
torchvision