# so we use pytorch 2.6
torch==2.6.0
torchvision==0.21.0
torchao==0.9.0

#torch>=2.4.0
#torchvision>=0.19.0

opencv-python>=4.9.0.80
diffusers==0.31.0
transformers>=4.49.0
tokenizers>=0.20.3
accelerate>=1.1.1

flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

tqdm
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
nvidia-tensorrt
onnx
onnxruntime
onnxscript
onnxconverter_common
flask
flask-socketio