From b16af90bc3948df608f20b36bb975a78b33504d6 Mon Sep 17 00:00:00 2001
From: kk <43161300+kkHuang-amd@users.noreply.github.com>
Date: Thu, 6 Mar 2025 19:38:54 +0800
Subject: [PATCH] AMD/ROCm: update base image string (#4137)

Co-authored-by: wunhuang <wunhuang@amd.com>
Co-authored-by: yichiche <yichiche@amd.com>
---
 docker/Dockerfile.rocm | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm
index ca62ab510..eb055f6cc 100644
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
@@ -2,7 +2,7 @@
 #   docker build --build-arg SGL_BRANCH=v0.4.3.post3 -t v0.4.3.post3-rocm630 -f Dockerfile.rocm .
 
 # default base image
-ARG BASE_IMAGE="rocm/vllm-dev:20250114"
+ARG BASE_IMAGE="rocm/sgl-dev:vllm20250114"
 
 FROM $BASE_IMAGE AS base
 USER root