plarnholt commited on
Commit
2848675
·
verified ·
1 Parent(s): 89e6e22

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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 Torch (CUDA 12.1)
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
- # Fix Outlines/Numba issue and its extra dependency
15
  outlines==0.0.46
16
  numba==0.59.1
17
- pyairports==2.1.1
 
 
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