From d2966af1f7c389dbdf18e5a7d2354667e7ed57df Mon Sep 17 00:00:00 2001
From: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Date: Wed, 18 Dec 2024 10:55:59 -0600
Subject: [PATCH] Using the generic base image created by the vllm-ci pipeline

---
 Dockerfile.rocm | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 1f44958f5d1f1..13b0a764cbd7e 100644
--- 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