Vintern-1B-v3.5-Demo / requirements.txt
fffiloni's picture
add required flash attention
8227cd8 verified
raw
history blame
181 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
flash-attn==2.5.8