numpy
Pillow
torch
torchvision
xformers
diffusers
transformers