chore: bump mooncake version to 0.3.10.post2 (#23439)
This commit is contained in:
@@ -457,10 +457,10 @@ fi
|
|||||||
# parse the first two digits of CU_VERSION (pytorch convention is cu{major}{minor}
|
# parse the first two digits of CU_VERSION (pytorch convention is cu{major}{minor}
|
||||||
# with a single-digit minor, e.g. cu126, cu129, cu130).
|
# with a single-digit minor, e.g. cu126, cu129, cu130).
|
||||||
if [ "$CU_MAJOR" = "13" ]; then
|
if [ "$CU_MAJOR" = "13" ]; then
|
||||||
MOONCAKE_PKG="mooncake-transfer-engine-cuda13==0.3.10.post1"
|
MOONCAKE_PKG="mooncake-transfer-engine-cuda13==0.3.10.post2"
|
||||||
EXTRA_NVIDIA_SPECS="nvidia-cuda-nvrtc"
|
EXTRA_NVIDIA_SPECS="nvidia-cuda-nvrtc"
|
||||||
else
|
else
|
||||||
MOONCAKE_PKG="mooncake-transfer-engine==0.3.10.post1"
|
MOONCAKE_PKG="mooncake-transfer-engine==0.3.10.post2"
|
||||||
EXTRA_NVIDIA_SPECS="nvidia-cuda-nvrtc-cu12"
|
EXTRA_NVIDIA_SPECS="nvidia-cuda-nvrtc-cu12"
|
||||||
fi
|
fi
|
||||||
$PIP_CMD install ${MOONCAKE_PKG} ${EXTRA_NVIDIA_SPECS} py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
|
$PIP_CMD install ${MOONCAKE_PKG} ${EXTRA_NVIDIA_SPECS} py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
|
||||||
|
|||||||
Reference in New Issue
Block a user