fix: add flash attn
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -13,4 +13,5 @@ requests
|
|
| 13 |
openai
|
| 14 |
PyYAML
|
| 15 |
einops
|
| 16 |
-
huggingface_hub
|
|
|
|
|
|
| 13 |
openai
|
| 14 |
PyYAML
|
| 15 |
einops
|
| 16 |
+
huggingface_hub
|
| 17 |
+
flash-attn
|