chore: bump mooncake version to 0.3.8.post1 (#16792)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
+1
-1
@@ -291,7 +291,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
if [ "$GRACE_BLACKWELL" = "1" ]; then \
|
if [ "$GRACE_BLACKWELL" = "1" ]; then \
|
||||||
python3 -m pip install mooncake-transfer-engine==0.3.7.post2; \
|
python3 -m pip install mooncake-transfer-engine==0.3.7.post2; \
|
||||||
else \
|
else \
|
||||||
python3 -m pip install mooncake-transfer-engine==0.3.8; \
|
python3 -m pip install mooncake-transfer-engine==0.3.8.post1; \
|
||||||
fi
|
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)
|
||||||
|
|||||||
@@ -76,8 +76,8 @@ def get_allocator_from_storage(allocator_type):
|
|||||||
return MooncakeHostTensorAllocator()
|
return MooncakeHostTensorAllocator()
|
||||||
except ImportError:
|
except ImportError:
|
||||||
logger.warning(
|
logger.warning(
|
||||||
"Mooncake's tensor allocator requires mooncake >= 0.3.8. "
|
"Mooncake's tensor allocator requires mooncake >= 0.3.8.post1. "
|
||||||
"Please upgrade Mooncake by 'pip install mooncake --upgrade'. "
|
"Please upgrade Mooncake by 'pip install mooncake-transfer-engine --upgrade'. "
|
||||||
"Fallback to use default allocator."
|
"Fallback to use default allocator."
|
||||||
)
|
)
|
||||||
return HostTensorAllocator()
|
return HostTensorAllocator()
|
||||||
|
|||||||
@@ -143,7 +143,7 @@ fi
|
|||||||
$PIP_CMD list
|
$PIP_CMD list
|
||||||
|
|
||||||
# Install other python dependencies
|
# 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
|
if [ "$IS_BLACKWELL" != "1" ]; then
|
||||||
# For lmms_evals evaluating MMMU
|
# For lmms_evals evaluating MMMU
|
||||||
|
|||||||
Reference in New Issue
Block a user