diff --git a/Dockerfile.mxfp4 b/Dockerfile.mxfp4 index 1902e32..e00184f 100644 --- a/Dockerfile.mxfp4 +++ b/Dockerfile.mxfp4 @@ -175,7 +175,9 @@ RUN --mount=type=cache,id=repo-cache,target=/repo-cache \ # 2. Logic: Clone if missing, otherwise Fetch & Reset if [ ! -d "vllm-mxfp4" ]; then \ echo "Cache miss: Cloning vLLM from scratch..." && \ - git clone --recursive ${VLLM_REPO} vllm-mxfp4; \ + git clone --recursive ${VLLM_REPO} vllm-mxfp4 && \ + cd vllm-mxfp4 && \ + git checkout ${VLLM_SHA} ; \ else \ echo "Cache hit: Fetching updates..." && \ cd vllm-mxfp4 && \ diff --git a/build-and-copy.sh b/build-and-copy.sh index 2030909..0b45eb2 100755 --- a/build-and-copy.sh +++ b/build-and-copy.sh @@ -11,7 +11,7 @@ REBUILD_VLLM=false COPY_HOSTS=() SSH_USER="$USER" NO_BUILD=false -TRITON_REF="v3.6.0" +TRITON_REF="v3.5.1" VLLM_REF="main" TMP_IMAGE="" PARALLEL_COPY=false