diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 9b1d67b5e..7e6ae193a 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -2,7 +2,7 @@ # docker build --build-arg SGL_BRANCH=v0.4.1.post5 -t v0.4.1.post5-rocm620 -f Dockerfile.rocm . # default base image -ARG BASE_IMAGE="rocmshared/vllm-rocm:20241031-tuned" +ARG BASE_IMAGE="rocmshared/vllm-rocm:20250113-tuned-elementwise" FROM $BASE_IMAGE AS base USER root