Spaces:
Running
on
Zero
Running
on
Zero
roychao19477
commited on
Commit
·
f515c0f
1
Parent(s):
37980af
Upload wheel
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -6,6 +6,7 @@ argparse
|
|
| 6 |
tensorboard
|
| 7 |
pesq
|
| 8 |
einops
|
|
|
|
| 9 |
numpy==1.26.4
|
| 10 |
torch==2.1
|
| 11 |
torchaudio==2.1
|
|
|
|
| 6 |
tensorboard
|
| 7 |
pesq
|
| 8 |
einops
|
| 9 |
+
flash_attn==2.5.7
|
| 10 |
numpy==1.26.4
|
| 11 |
torch==2.1
|
| 12 |
torchaudio==2.1
|