torch==2.0.1
torchvision==0.15.2
numpy
scipy
scikit-image
opencv-python
pillow
diffusers==0.20.2
transformers
xformers==0.0.20
accelerate
matplotlib
tqdm
gradio==3.44.4
einops
onnxruntime
basicsr
av
cloudpickle
fvcore
omegaconf==2.1
hydra-core
pycocotools
pytorch-lightning==1.5.0