einops
gradio
numpy
torch
torchvision
pytorch-lightning
diffusers
transformers
accelerate==0.31.0
opencv-python