requirements.txt

modelscope==1.8.4
xformers==0.0.20
torch==2.0.1
open_clip_torch>=2.0.2
opencv-python-headless
opencv-python 
einops>=0.4
rotary-embedding-torch
fairscale 
scipy
imageio
pytorch-lightning
torchsde
easydict