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