Skip to content

GitLab

Unverified Commit 0f334945 authored by shangmingc's avatar shangmingc Committed by GitHub
Browse files

[CI] Fix PD mooncake dependency error (#6212) 


Signed-off-by: default avatarShangming Cai <caishangming@linux.alibaba.com>
parent fba8eccd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/ci_install_dependency.sh
View file @ 0f334945
...@@ -20,7 +20,7 @@ pip install -e "python[all]" ...@@ -20,7 +20,7 @@ pip install -e "python[all]"
# Install additional dependencies # Install additional dependencies
pip install torch_memory_saver 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 # For compiling xgrammar kernels
pip install cuda-python nvidia-cuda-nvrtc-cu12 pip install cuda-python nvidia-cuda-nvrtc-cu12
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment