Spaces:
Build error
Build error
Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
torch
|
| 2 |
transformers
|
| 3 |
-
|
| 4 |
einops
|
|
|
|
| 1 |
+
torch==2.2.0
|
| 2 |
transformers
|
| 3 |
+
flash-attn==2.2.0
|
| 4 |
einops
|