[Fix]Pin mooncake version to 0.3.7.post2 in grace blackwell (#16502)

This commit is contained in:
Baizhou Zhang
2026-01-06 14:11:40 +08:00
committed by GitHub
parent 73398e22d6
commit 6ffe1fc02f

View File

@@ -273,7 +273,6 @@ RUN --mount=type=cache,target=/root/.cache/pip \
RUN --mount=type=cache,target=/root/.cache/pip \
python3 -m pip install \
datamodel_code_generator \
mooncake-transfer-engine==0.3.8 \
pre-commit \
pytest \
black \
@@ -286,6 +285,14 @@ RUN --mount=type=cache,target=/root/.cache/pip \
py-spy \
cubloaty
# Install mooncake
RUN --mount=type=cache,target=/root/.cache/pip \
if [ "$GRACE_BLACKWELL" = "1" ]; then \
python3 -m pip install mooncake-transfer-engine==0.3.7.post2; \
else \
python3 -m pip install mooncake-transfer-engine==0.3.8; \
fi
# Build and install sgl-model-gateway (install Rust, build, then remove to save space)
RUN --mount=type=cache,target=/root/.cache/pip \
curl --proto '=https' --tlsv1.2 --retry 3 --retry-delay 2 -sSf https://sh.rustup.rs | sh -s -- -y \