Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
825d00e
1
Parent(s):
5f56a57
feat: add flash-attn prebuilt wheel to requirements
Browse files- requirements.txt +3 -0
requirements.txt
CHANGED
|
@@ -10,3 +10,6 @@ torchaudio
|
|
| 10 |
peft
|
| 11 |
scipy
|
| 12 |
backoff
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
peft
|
| 11 |
scipy
|
| 12 |
backoff
|
| 13 |
+
|
| 14 |
+
# flash-attn (prebuilt)
|
| 15 |
+
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|