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

Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>
This commit is contained in:
shangmingc
2025-05-13 01:08:49 +08:00
committed by GitHub
parent fba8eccd7e
commit 0f334945c6

View File

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