[Fix]Pin mooncake version to 0.3.7.post2 in grace blackwell (#16502)
This commit is contained in:
+8
-1
@@ -273,7 +273,6 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
python3 -m pip install \
|
python3 -m pip install \
|
||||||
datamodel_code_generator \
|
datamodel_code_generator \
|
||||||
mooncake-transfer-engine==0.3.8 \
|
|
||||||
pre-commit \
|
pre-commit \
|
||||||
pytest \
|
pytest \
|
||||||
black \
|
black \
|
||||||
@@ -286,6 +285,14 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
py-spy \
|
py-spy \
|
||||||
cubloaty
|
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)
|
# Build and install sgl-model-gateway (install Rust, build, then remove to save space)
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
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 \
|
curl --proto '=https' --tlsv1.2 --retry 3 --retry-delay 2 -sSf https://sh.rustup.rs | sh -s -- -y \
|
||||||
|
|||||||
Reference in New Issue
Block a user