Vintern-1B-v3.5-Demo / requirements.txt
fffiloni's picture
Update requirements.txt
b286df8 verified
raw
history blame
305 Bytes
torch==2.3.1
torchvision==0.18.1
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.12.0
https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.3cxx11abiTRUE-cp310-cp310-linux_x86_64.whl