Hugging Face's logo

Spaces:
fffiloni
/
Sa2VA-simple-demo
Running on Zero

App Files Community
fffiloni commited on
Commit
db1df47
·
verified ·
1 Parent(s): 6708b3a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch==2.3.1
2
- torchvision==0.18.1
3
  transformers
4
  xtuner[deepspeed]==0.1.23
5
  timm==1.0.9
@@ -8,4 +8,4 @@ hydra-core==1.3.2
8
  ninja==1.11.1
9
  decord==0.6.0
10
  peft==0.12.0
11
- 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
 
1
+ torch==2.1.2
2
+ torchvision
3
  transformers
4
  xtuner[deepspeed]==0.1.23
5
  timm==1.0.9
 
8
  ninja==1.11.1
9
  decord==0.6.0
10
  peft==0.12.0
11
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu118torch2.1cxx11abiTRUE-cp310-cp310-linux_x86_64.whl