Update base image for ROCm (#2852)

Co-authored-by: HAI <hixiao@gmail.com>

Update base image for ROCm (#2852)
Co-authored-by: HAI <hixiao@gmail.com>
a18ab81d · sogalin · GitHub · 0bb0f763 · a18ab81d
Unverified Commit a18ab81d authored Jan 13, 2025 by sogalin Committed by GitHub Jan 13, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docker/Dockerfile.rocm docker/Dockerfile.rocm +1 -1

No files found.
--- 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