Hugging Face's logo

Spaces:
shenyunhang
/
Long-VITA
Running on Zero

App Files Community
shenyunhang commited on
Commit
66548c3
·
verified ·
1 Parent(s): c33ee23

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch==2.6.0
2
  torchvision
3
  #flash-attn
4
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
5
  deepspeed
6
  accelerate
7
  transformers
 
1
+ torch==2.4.0
2
  torchvision
3
  #flash-attn
4
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
5
  deepspeed
6
  accelerate
7
  transformers