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