Update base image for ROCm (#2852)
Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
@@ -2,7 +2,7 @@
|
|||||||
# docker build --build-arg SGL_BRANCH=v0.4.1.post5 -t v0.4.1.post5-rocm620 -f Dockerfile.rocm .
|
# docker build --build-arg SGL_BRANCH=v0.4.1.post5 -t v0.4.1.post5-rocm620 -f Dockerfile.rocm .
|
||||||
|
|
||||||
# default base image
|
# 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
|
FROM $BASE_IMAGE AS base
|
||||||
USER root
|
USER root
|
||||||
|
|||||||
Reference in New Issue
Block a user