[Fix]Pin mooncake version to 0.3.7.post2 in grace blackwell (#16502)
This commit is contained in:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user