torch
torchvision
einops
pytorch-lightning
omegaconf
open-clip-torch
xformers
taming-transformers
gradio
Pillow
numpy
spaces