torch
torchvision
transformers
diffusers
matplotlib