Update base image for ROCm (#2852)

Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
sogalin
2025-01-13 14:39:44 +08:00
committed by GitHub
parent 0bb0f76311
commit a18ab81ddd

View File

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