xfcxcxcdfdfd commited on
Commit
8f73f76
·
verified ·
1 Parent(s): 4e9a90d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -83,7 +83,7 @@ RUN pip install --upgrade pip && \
83
  pip install --no-cache-dir --pre torch==2.6.0.dev20241122 --index-url https://download.pytorch.org/whl/nightly/rocm6.2
84
 
85
  # Install vllm and huggingface-hub
86
- RUN pip install vllm && \
87
  pip install huggingface-hub[cli]
88
 
89
  # Expose port for the model server
 
83
  pip install --no-cache-dir --pre torch==2.6.0.dev20241122 --index-url https://download.pytorch.org/whl/nightly/rocm6.2
84
 
85
  # Install vllm and huggingface-hub
86
+ RUN pip install vllm==0.6.2 && \
87
  pip install huggingface-hub[cli]
88
 
89
  # Expose port for the model server