diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh index 591844686..139331578 100755 --- a/scripts/ci_install_dependency.sh +++ b/scripts/ci_install_dependency.sh @@ -20,7 +20,7 @@ pip install flashinfer_python==0.2.3 --find-links ${FLASHINFER_REPO} --force-rei pip install torch_memory_saver --force-reinstall -pip install transformers==4.45.2 sentence_transformers accelerate peft pandas datasets +pip install transformers==4.45.2 sentence_transformers accelerate==1.4.0 peft pandas datasets # For compling xgrammar kernels pip install cuda-python nvidia-cuda-nvrtc-cu12