chore: bump mooncake version to 0.3.8.post1 (#16792)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
@@ -291,7 +291,7 @@ 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; \
|
||||
python3 -m pip install mooncake-transfer-engine==0.3.8.post1; \
|
||||
fi
|
||||
|
||||
# Build and install sgl-model-gateway (install Rust, build, then remove to save space)
|
||||
|
||||
@@ -76,8 +76,8 @@ def get_allocator_from_storage(allocator_type):
|
||||
return MooncakeHostTensorAllocator()
|
||||
except ImportError:
|
||||
logger.warning(
|
||||
"Mooncake's tensor allocator requires mooncake >= 0.3.8. "
|
||||
"Please upgrade Mooncake by 'pip install mooncake --upgrade'. "
|
||||
"Mooncake's tensor allocator requires mooncake >= 0.3.8.post1. "
|
||||
"Please upgrade Mooncake by 'pip install mooncake-transfer-engine --upgrade'. "
|
||||
"Fallback to use default allocator."
|
||||
)
|
||||
return HostTensorAllocator()
|
||||
|
||||
@@ -143,7 +143,7 @@ fi
|
||||
$PIP_CMD list
|
||||
|
||||
# Install other python dependencies
|
||||
$PIP_CMD install mooncake-transfer-engine==0.3.8 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
|
||||
$PIP_CMD install mooncake-transfer-engine==0.3.8.post1 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
|
||||
|
||||
if [ "$IS_BLACKWELL" != "1" ]; then
|
||||
# For lmms_evals evaluating MMMU
|
||||
|
||||
Reference in New Issue
Block a user