fix followup #2517

sgl-project · Dec 19, 2024 · 2a458ac · 2a458ac
1 parent 4b83db2
commit 2a458ac
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