Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -3,4 +3,5 @@ tiktoken==0.6.0
|
|
| 3 |
torch
|
| 4 |
pillow
|
| 5 |
accelerate>=0.26.0
|
| 6 |
-
fastapi==0.112.2
|
|
|
|
|
|
| 3 |
torch
|
| 4 |
pillow
|
| 5 |
accelerate>=0.26.0
|
| 6 |
+
fastapi==0.112.2
|
| 7 |
+
flash-attn==2.5.7
|