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
...@@ -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