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

Use cu128 for torch audio to fix some CI tests (#11251)

Browse Source
This commit is contained in:
Lianmin Zheng
2025-10-05 19:52:32 -07:00
committed by GitHub
parent baee08601b
commit 366a603e95
3 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/ci/ci_install_dependency.sh
View File

@@ -24,7 +24,7 @@ if [ "$IS_BLACKWELL" = "1" ]; then
PIP_INSTALL_SUFFIX="--break-system-packages"
# Clean up existing installations
$PIP_CMD uninstall -y flashinfer_python sgl-kernel sglang vllm torch $PIP_INSTALL_SUFFIX || true
$PIP_CMD uninstall -y flashinfer_python sgl-kernel sglang vllm torch torchaudio $PIP_INSTALL_SUFFIX || true
else
# In normal cases, we use uv, which is much faster than pip.
pip install --upgrade pip
@@ -35,7 +35,7 @@ else
PIP_INSTALL_SUFFIX="--index-strategy unsafe-best-match"
# Clean up existing installations
$PIP_CMD uninstall flashinfer_python sgl-kernel sglang vllm torch || true
$PIP_CMD uninstall flashinfer_python sgl-kernel sglang vllm torch torchaudio || true
fi
# Install the main package