diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index b186f88d2744..34641dd0a898 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -28,7 +28,8 @@ ARG VLLM_REPO="https://github.com/vllm-project/vllm.git" ARG VLLM_BRANCH="main" ONBUILD RUN git clone ${VLLM_REPO} \ && cd vllm \ - && git checkout ${VLLM_BRANCH} + && git fetch -v --prune -- origin ${VLLM_BRANCH} \ + && git checkout FETCH_HEAD FROM fetch_vllm_${REMOTE_VLLM} AS fetch_vllm # -----------------------