chore: upgrade mooncake 0.3.6.post1 to fix gb200 dockerfile (#10681)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
2
.github/workflows/pr-test-pd-router.yml
vendored
2
.github/workflows/pr-test-pd-router.yml
vendored
@@ -140,7 +140,7 @@ jobs:
|
|||||||
python3 -m pip --no-cache-dir install --upgrade pip
|
python3 -m pip --no-cache-dir install --upgrade pip
|
||||||
python3 -m pip --no-cache-dir install torch==2.8.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu126
|
python3 -m pip --no-cache-dir install torch==2.8.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu126
|
||||||
python3 -m pip --no-cache-dir install -e "python[all]" --break-system-packages
|
python3 -m pip --no-cache-dir install -e "python[all]" --break-system-packages
|
||||||
python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.6
|
python3 -m pip --no-cache-dir install mooncake-transfer-engine==0.3.6.post1
|
||||||
python3 -m pip --no-cache-dir install --user --force-reinstall genai-bench==0.0.2
|
python3 -m pip --no-cache-dir install --user --force-reinstall genai-bench==0.0.2
|
||||||
|
|
||||||
- name: Build and install sgl-router
|
- name: Build and install sgl-router
|
||||||
|
|||||||
@@ -128,7 +128,7 @@ RUN cd /sgl-workspace/DeepEP && \
|
|||||||
# Python tools
|
# Python tools
|
||||||
RUN python3 -m pip install --no-cache-dir \
|
RUN python3 -m pip install --no-cache-dir \
|
||||||
datamodel_code_generator \
|
datamodel_code_generator \
|
||||||
mooncake-transfer-engine==0.3.6 \
|
mooncake-transfer-engine==0.3.6.post1 \
|
||||||
pre-commit \
|
pre-commit \
|
||||||
pytest \
|
pytest \
|
||||||
black \
|
black \
|
||||||
|
|||||||
@@ -118,7 +118,7 @@ RUN cd /sgl-workspace/DeepEP && \
|
|||||||
# Python tools
|
# Python tools
|
||||||
RUN python3 -m pip install --no-cache-dir \
|
RUN python3 -m pip install --no-cache-dir \
|
||||||
datamodel_code_generator \
|
datamodel_code_generator \
|
||||||
mooncake-transfer-engine==0.3.6 \
|
mooncake-transfer-engine==0.3.6.post1 \
|
||||||
pre-commit \
|
pre-commit \
|
||||||
pytest \
|
pytest \
|
||||||
black \
|
black \
|
||||||
|
|||||||
@@ -74,7 +74,7 @@ fi
|
|||||||
$PIP_CMD list
|
$PIP_CMD list
|
||||||
|
|
||||||
# Install additional dependencies
|
# Install additional dependencies
|
||||||
$PIP_CMD install mooncake-transfer-engine==0.3.6 nvidia-cuda-nvrtc-cu12 py-spy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX
|
$PIP_CMD install mooncake-transfer-engine==0.3.6.post1 nvidia-cuda-nvrtc-cu12 py-spy huggingface_hub[hf_xet] $PIP_INSTALL_SUFFIX
|
||||||
|
|
||||||
if [ "$IS_BLACKWELL" != "1" ]; then
|
if [ "$IS_BLACKWELL" != "1" ]; then
|
||||||
# For lmms_evals evaluating MMMU
|
# For lmms_evals evaluating MMMU
|
||||||
|
|||||||
Reference in New Issue
Block a user