Using the generic base image created by the vllm-ci pipeline (ROCm#336)

gshtras · web-flow · commit 399016d72238 · 2024-12-18T13:41:09.000-05:00
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -3,7 +3,7 @@ ARG REMOTE_VLLM="0"
 ARG USE_CYTHON="0"
 ARG BUILD_RPD="1"
 ARG COMMON_WORKDIR=/app
-ARG BASE_IMAGE=rocm/vllm-dev:base_ubuntu22.04_py3.12_ROCm6.3_hipblaslt0.12_torch2.6
+ARG BASE_IMAGE=rocm/vllm-dev:base
 
 FROM ${BASE_IMAGE} AS base
 
