diff --git a/Dockerfile.mxfp4 b/Dockerfile.mxfp4 index 00c6968..2c79cd5 100644 --- a/Dockerfile.mxfp4 +++ b/Dockerfile.mxfp4 @@ -101,7 +101,7 @@ ARG FLASHINFER_SHA=f349e52496a72a00d8c4ac02c7a1e38523ff7194 ARG CUTLASS_SHA=fede53000a962b46e05bafe0c86311778caeb380 RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ - uv pip install nvidia-nvshmem-cu13 "apache-tvm-ffi<0.2" + uv pip install "nvidia-nvshmem-cu13<3.6" "apache-tvm-ffi<0.2" # Clone FlashInfer (cached for faster rebuilds) RUN --mount=type=cache,id=git-flashinfer,target=/git-cache/flashinfer \ @@ -272,7 +272,7 @@ ENV PATH=$VLLM_BASE_DIR:$PATH # Final extra deps RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ - uv pip install ray[default] fastsafetensors nvidia-nvshmem-cu13 + uv pip install ray[default] fastsafetensors "nvidia-nvshmem-cu13<3.6" # Build metadata (generated by build-and-copy.sh) COPY build-metadata.yaml /workspace/build-metadata.yaml