opencv-python>=4.9.0.80
diffusers==0.31.0
transformers>=4.49.0,<5
tokenizers>=0.20.3
accelerate>=1.1.1
tqdm
datasets
imageio
easydict
ftfy
dashscope
imageio-ffmpeg
# numpy==1.24.4
wandb
omegaconf
einops
av==13.1.0
opencv-python
git+https://github.com/openai/CLIP.git
open_clip_torch
starlette
pycocotools
lmdb
matplotlib
sentencepiece
pydantic==2.10.6
scikit-image
huggingface_hub[cli]
dominate
# Optional TensorRT support. Install separately because nvidia-tensorrt is a
# placeholder on PyPI and requires NVIDIA's package index to be configured first:
#   pip install nvidia-pyindex
#   pip install nvidia-tensorrt
#   pip install pycuda
onnx
onnxruntime
onnxscript
onnxconverter_common
flask
flask-socketio
torchao==0.13.0  # paired with the documented PyTorch 2.8.0 environment
peft