Spaces:
Runtime error
Runtime error
Commit
•
08b4361
1
Parent(s):
7091430
add flash attn
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -2,4 +2,5 @@
|
|
2 |
torch
|
3 |
pip install git+https://github.com/huggingface/transformers
|
4 |
accelerate
|
5 |
-
optimum
|
|
|
|
2 |
torch
|
3 |
pip install git+https://github.com/huggingface/transformers
|
4 |
accelerate
|
5 |
+
optimum
|
6 |
+
flash-attn --no-build-isolation
|