Skip to content

Commit

Permalink
[REMOVE ME] Bump FlashInfer version for CI tests
Browse files Browse the repository at this point in the history
  • Loading branch information
james-p-xu committed Nov 16, 2024
1 parent d268aca commit 0015a72
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion scripts/ci_install_dependency.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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 .

0 comments on commit 0015a72

Please sign in to comment.