requirements.txt

torch==2.3.1 -f https://download.pytorch.org/whl/cu124/torch_stable.html
torchvision==0.18.1 -f https://download.pytorch.org/whl/cu124/torch_stable.html
diffusers==0.32.0
transformers==4.42.3
xtuner[deepspeed]==0.1.23
timm==1.0.9
mmdet==3.3.0
hydra-core==1.3.2
ninja==1.11.1
decord==0.6.0
peft==0.11.1
protobuf==5.29.4
sentencepiece==0.2.0
tornado==6.4.2
openai==0.28.0
gradio==5.32.0
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
opencv-python