pvduy commited on
Commit
1a94c2b
·
verified ·
1 Parent(s): 8c23b1a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -8,8 +8,8 @@ RUN apt-get update -y \
8
  # this won't be needed for future versions of this docker image
9
  # or future versions of triton.
10
  RUN ldconfig /usr/local/cuda-12.1/compat/
11
- RUN pip install cmake>=3.21 ninja psutil sentencepiece numpy requests py-cpuinfo transformers >= 4.39.1 fastapi uvicorn[standard] pydantic >= 2.0 prometheus_client >= 0.18.0 tiktoken == 0.6.0 outlines == 0.0.34
12
- RUN ray >= 2.9 pynvml==11.5.0 vllm-nccl-cu12>=2.18,<2.19 torch==2.2.1 xformers == 0.0.25 triton>=2.1.0
13
 
14
  WORKDIR /code
15
 
 
8
  # this won't be needed for future versions of this docker image
9
  # or future versions of triton.
10
  RUN ldconfig /usr/local/cuda-12.1/compat/
11
+ RUN pip install cmake>=3.21 ninja psutil sentencepiece numpy requests py-cpuinfo transformers>=4.39.1 fastapi uvicorn[standard] pydantic>=2.0 prometheus_client>=0.18.0 tiktoken==0.6.0 outlines==0.0.34
12
+ RUN ray>=2.9 pynvml==11.5.0 vllm-nccl-cu12>=2.18,<2.19 torch==2.2.1 xformers==0.0.25 triton>=2.1.0
13
 
14
  WORKDIR /code
15