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