sofianhw commited on
Commit
72f70b3
1 Parent(s): 5491dc5

fix llama 3.1

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -14,7 +14,7 @@ RUN pip3 install "torch==2.1.1"
14
  # This build is slow but NVIDIA does not provide binaries. Increase MAX_JOBS as needed.
15
  # RUN pip3 install "git+https://github.com/stanford-futuredata/megablocks.git"
16
  RUN pip3 install -U openai
17
- RUN pip3 install vllm==0.5.3
18
  RUN pip3 install -U pydantic
19
  RUN pip3 install -U aioprometheus
20
 
 
14
  # This build is slow but NVIDIA does not provide binaries. Increase MAX_JOBS as needed.
15
  # RUN pip3 install "git+https://github.com/stanford-futuredata/megablocks.git"
16
  RUN pip3 install -U openai
17
+ RUN pip3 install vllm==0.5.3.post1
18
  RUN pip3 install -U pydantic
19
  RUN pip3 install -U aioprometheus
20