minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[NPU] fix CI (#26390)

Browse Source
This commit is contained in:
Makcum888e
2026-05-27 09:45:42 +03:00
committed by GitHub
parent f32ca1e0e5
commit 9060509214
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/pyproject_npu.toml
-1
View File
@@ -57,7 +57,6 @@ dependencies = [
"tiktoken", "tiktoken",
"timm==1.0.16", "timm==1.0.16",
"torchao==0.9.0", "torchao==0.9.0",
"torchaudio==2.8.0",
"tqdm", "tqdm",
"mistral_common>=1.11.0", "mistral_common>=1.11.0",
"transformers==5.8.1", "transformers==5.8.1",
scripts/ci/npu/npu_ci_install_dependency.sh
+2 -1
View File
@@ -57,7 +57,8 @@ if [ -n "$OPTIONAL_DEPS" ]; then
else else
PYTORCH_VERSION="2.8.0" PYTORCH_VERSION="2.8.0"
TORCHVISION_VERSION="0.23.0" TORCHVISION_VERSION="0.23.0"
${UV_PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} --index-url ${TORCH_CACHE_URL:="https://download.pytorch.org/whl/cpu"} --extra-index-url ${PYPI_CACHE_URL:="https://pypi.org/simple/"} TORCHAUDIO_VERSION="2.8.0"
${UV_PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url ${TORCH_CACHE_URL:="https://download.pytorch.org/whl/cpu"} --extra-index-url ${PYPI_CACHE_URL:="https://pypi.org/simple/"}
PTA_URL="https://gitcode.com/Ascend/pytorch/releases/download/v7.3.0-pytorch2.8.0/torch_npu-2.8.0.post2-cp311-cp311-manylinux_2_28_aarch64.whl" PTA_URL="https://gitcode.com/Ascend/pytorch/releases/download/v7.3.0-pytorch2.8.0/torch_npu-2.8.0.post2-cp311-cp311-manylinux_2_28_aarch64.whl"
${PIP_INSTALL} ${PTA_URL} ${PIP_INSTALL} ${PTA_URL}
fi fi