Pin vllm package version

stackhpc · Dec 12, 2023 · 6876068 · 6876068
1 parent 007fd0b
commit 6876068
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/images/api-base/Dockerfile b/images/api-base/Dockerfile
@@ -1,3 +1,3 @@
 FROM nvidia/cuda:12.2.2-runtime-ubi8
 RUN dnf install -y python3.11-pip && dnf clean all
-RUN python3.11 -m pip install --no-cache-dir vllm && python3.11 -m pip cache purge
+RUN python3.11 -m pip install --no-cache-dir vllm==0.2.4 && python3.11 -m pip cache purge