From 2a458acc37a301b008d64d739ed511cd4412e3df Mon Sep 17 00:00:00 2001
From: zhyncs <me@zhyncs.com>
Date: Thu, 19 Dec 2024 23:15:28 +0800
Subject: [PATCH] fix followup #2517

---
 scripts/ci_install_dependency.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh
index 325da2b5c2..8fb93a3e82 100755
--- 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