Long-VITA / requirements.txt
shenyunhang's picture
Update requirements.txt
e592747 verified
raw
history blame
289 Bytes
torch==2.4.0
torchvision
#flash-attn
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
deepspeed
accelerate
transformers
timm
moviepy
decord
pillow
opencv-python
filetype
natsort
gradio