diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh index fd0299db05e..cd2ed4ccd9f 100644 --- a/scripts/ci_install_dependency.sh +++ b/scripts/ci_install_dependency.sh @@ -5,4 +5,6 @@ Install the dependency in CI. pip install --upgrade pip pip install -e "python[all]" pip install transformers==4.45.2 sentence_transformers accelerate peft -pip install flashinfer -i https://flashinfer.ai/whl/cu121/torch2.4/ --force-reinstall +git clone https://github.com/flashinfer-ai/flashinfer.git --recursive +cd flashinfer/python +pip install -e .