Update Dockerfile.rocm

vllm-project · Dec 20, 2024 · b4023cd · b4023cd
1 parent 47a0b61
commit b4023cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -171,4 +171,5 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 # install development dependencies (for testing)
 RUN python3 -m pip install -e tests/vllm_test_utils
 
-CMD ["/bin/bash"]
+ENTRYPOINT ["python3", "-m", "vllm.entrypoints.openai.api_server"]
+