fffiloni commited on
Commit
8227cd8
·
verified ·
1 Parent(s): da27b55

add required flash attention

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -7,4 +7,5 @@ mmdet==3.3.0
7
  hydra-core==1.3.2
8
  ninja==1.11.1
9
  decord==0.6.0
10
- peft==0.12.0
 
 
7
  hydra-core==1.3.2
8
  ninja==1.11.1
9
  decord==0.6.0
10
+ peft==0.12.0
11
+ flash-attn==2.5.8