Spaces:
Running
on
Zero
Running
on
Zero
add required flash attention
Browse files- 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
|