From 399016d722381845bbf7e1507839f72c67cc72e4 Mon Sep 17 00:00:00 2001
From: Gregory Shtrasberg <156009573+gshtras@users.noreply.github.com>
Date: Wed, 18 Dec 2024 13:41:09 -0500
Subject: [PATCH] Using the generic base image created by the vllm-ci pipeline
 (#336)

---
 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