imageio==2.33.1
imageio-ffmpeg==0.4.9
matplotlib==3.8.2
matplotlib-inline==0.1.6
opencv-python==4.5.3.56
opencv-python-headless==4.9.0.80
pillow==10.2.0

torch==2.0.1
torchvision==0.15.2
torchsde==0.2.6

onnxruntime==1.18.0
# onnxruntime-gpu==1.13.1
open-clip-torch==2.24.0
ptflops==0.7.2.2
rotary-embedding-torch==0.5.3
thop==0.1.1.post2209072238
xformers==0.0.20

# pynvml==11.5.0
oss2==2.18.4
PyYAML==6.0.1
requests==2.31.0
simplejson==3.19.2
tqdm==4.66.1
typing-inspect==0.9.0
typing_extensions==4.9.0

transformers==4.48.2
safetensors==0.5.2
diffusers==0.32.2

einops==0.7.0
huggingface-hub==0.25.0
gradio==5.15.0