diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh index b496ac787..c187bf81b 100755 --- a/scripts/ci_install_dependency.sh +++ b/scripts/ci_install_dependency.sh @@ -26,4 +26,4 @@ pip install transformers==4.45.2 sentence_transformers accelerate peft pandas da pip install cuda-python nvidia-cuda-nvrtc-cu12 # reinstall sgl-kernel -pip install sgl-kernel --force-reinstall --no-deps +pip install sgl-kernel==0.0.3.post6 --force-reinstall --no-deps