[CI] Upgrade mooncake to v0.3.4.post2 to fix potential slice failed bug (#7522)
Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>
This commit is contained in:
@@ -85,7 +85,7 @@ RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.2.5/sour
|
||||
# Python tools
|
||||
RUN python3 -m pip install --no-cache-dir \
|
||||
datamodel_code_generator \
|
||||
mooncake_transfer_engine==0.3.4.post1 \
|
||||
mooncake_transfer_engine==0.3.4.post2 \
|
||||
pre-commit \
|
||||
pytest \
|
||||
black \
|
||||
|
||||
@@ -23,7 +23,7 @@ pip install -e "python[dev]"
|
||||
pip list
|
||||
|
||||
# Install additional dependencies
|
||||
pip install mooncake-transfer-engine==0.3.4.post1 nvidia-cuda-nvrtc-cu12
|
||||
pip install mooncake-transfer-engine==0.3.4.post2 nvidia-cuda-nvrtc-cu12
|
||||
|
||||
# For lmms_evals evaluating MMMU
|
||||
git clone --branch v0.3.3 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
|
||||
|
||||
Reference in New Issue
Block a user