Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -2,4 +2,5 @@ transformers==4.40.0
|
|
| 2 |
datasets
|
| 3 |
pillow
|
| 4 |
numpy
|
| 5 |
-
torch
|
|
|
|
|
|
| 2 |
datasets
|
| 3 |
pillow
|
| 4 |
numpy
|
| 5 |
+
torch
|
| 6 |
+
flash-attn
|