Spaces:
Paused
Paused
Update requirements.txt
Browse files- requirements.txt +4 -3
requirements.txt
CHANGED
|
@@ -3,7 +3,7 @@ uvicorn[standard]>=0.30
|
|
| 3 |
gradio>=4.38
|
| 4 |
requests>=2.31
|
| 5 |
|
| 6 |
-
# vLLM
|
| 7 |
vllm==0.5.5
|
| 8 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 9 |
torch==2.4.0
|
|
@@ -11,7 +11,8 @@ torch==2.4.0
|
|
| 11 |
transformers>=4.44
|
| 12 |
accelerate>=0.30
|
| 13 |
|
| 14 |
-
#
|
| 15 |
outlines==0.0.46
|
| 16 |
numba==0.59.1
|
| 17 |
-
|
|
|
|
|
|
| 3 |
gradio>=4.38
|
| 4 |
requests>=2.31
|
| 5 |
|
| 6 |
+
# vLLM + CUDA 12.1 compatible Torch
|
| 7 |
vllm==0.5.5
|
| 8 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 9 |
torch==2.4.0
|
|
|
|
| 11 |
transformers>=4.44
|
| 12 |
accelerate>=0.30
|
| 13 |
|
| 14 |
+
# Structured outputs stack used by vLLM<=0.6
|
| 15 |
outlines==0.0.46
|
| 16 |
numba==0.59.1
|
| 17 |
+
pycountry==24.6.1
|
| 18 |
+
pyairports==0.0.1
|