fix followup sgl-project#2517 (sgl-project#2524)

chosen-ox · Dec 22, 2024 · da9dbd0 · da9dbd0
1 parent 59fdd46
commit da9dbd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh
@@ -10,7 +10,7 @@ pip install --upgrade pip
 pip install -e "python[all]" --find-links https://flashinfer.ai/whl/cu121/torch2.4/flashinfer/
 
 # Force reinstall flashinfer
-pip install flashinfer --find-links ${FLASHINFER_REPO} --force-reinstall --no-deps
+pip install flashinfer==0.1.6 --find-links ${FLASHINFER_REPO} --force-reinstall --no-deps
 
 pip install transformers==4.45.2 sentence_transformers accelerate peft