requirements.txt

easydict==1.10
einops==0.6.1
fairscale==0.4.13
gradio==3.41.2
huggingface_hub==0.16.4
imageio==2.31.1
modelscope==1.8.4
open_clip_torch==2.20.0
opencv-python-headless==4.8.0.76
pytorch-lightning==2.0.7
rotary-embedding-torch==0.2.7
scipy==1.11.2
torch==2.0.1
torchsde==0.2.5
xformers==0.0.20