Update requirements.txt
Browse files- requirements.txt +5 -3
requirements.txt
CHANGED
|
@@ -2,6 +2,8 @@ fastapi==0.74.0
|
|
| 2 |
requests==2.27.1
|
| 3 |
uvicorn[standard]==0.17.6
|
| 4 |
sentencepiece==0.1.96
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
|
|
|
|
|
|
|
|
| 2 |
requests==2.27.1
|
| 3 |
uvicorn[standard]==0.17.6
|
| 4 |
sentencepiece==0.1.96
|
| 5 |
+
numpy==1.24.4 # Ensure compatibility
|
| 6 |
+
flash_attn==2.5.8
|
| 7 |
+
torch==2.3.1
|
| 8 |
+
accelerate==0.31.0
|
| 9 |
+
transformers==4.43.0
|