numpy
torch
torchvision
diffusers
transformers
tokenizers