EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[CI] Balance CI tests (#2293)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-12-01 01:47:30 -08:00
committed by GitHub
parent 5f12f0e7af
commit 9449a95431
3 changed files with 19 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/ci_install_dependency.sh
View File

@@ -6,8 +6,14 @@ Install the dependency in CI.
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 -i https://flashinfer.ai/whl/cu121/torch2.4/ --force-reinstall
pip install transformers==4.45.2 sentence_transformers accelerate peft
# for compling eagle kernels
# For compling eagle kernels
pip install cutex
# for compling xgrammar kernels
# For compling xgrammar kernels
pip install cuda-python nvidia-cuda-nvrtc-cu12