From 0f334945c644c843236d53832fbe45333ddc2e17 Mon Sep 17 00:00:00 2001 From: shangmingc Date: Tue, 13 May 2025 01:08:49 +0800 Subject: [PATCH] [CI] Fix PD mooncake dependency error (#6212) Signed-off-by: Shangming Cai --- scripts/ci_install_dependency.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh index 7344d1f86..9bd292a01 100755 --- a/scripts/ci_install_dependency.sh +++ b/scripts/ci_install_dependency.sh @@ -20,7 +20,7 @@ pip install -e "python[all]" # Install additional dependencies pip install torch_memory_saver -pip install transformers==4.51.0 timm torchaudio==2.6.0 sentence_transformers accelerate peft pandas datasets mooncake-transfer-engine +pip install transformers==4.51.0 timm torchaudio==2.6.0 sentence_transformers accelerate peft pandas datasets mooncake-transfer-engine==0.3.0 # For compiling xgrammar kernels pip install cuda-python nvidia-cuda-nvrtc-cu12