diff --git a/python/pyproject_npu.toml b/python/pyproject_npu.toml index 475756345..6fb3b95d6 100644 --- a/python/pyproject_npu.toml +++ b/python/pyproject_npu.toml @@ -57,7 +57,6 @@ dependencies = [ "tiktoken", "timm==1.0.16", "torchao==0.9.0", - "torchaudio==2.8.0", "tqdm", "mistral_common>=1.11.0", "transformers==5.8.1", diff --git a/scripts/ci/npu/npu_ci_install_dependency.sh b/scripts/ci/npu/npu_ci_install_dependency.sh index 9782861f4..83434b9ce 100755 --- a/scripts/ci/npu/npu_ci_install_dependency.sh +++ b/scripts/ci/npu/npu_ci_install_dependency.sh @@ -57,7 +57,8 @@ if [ -n "$OPTIONAL_DEPS" ]; then else PYTORCH_VERSION="2.8.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" ${PIP_INSTALL} ${PTA_URL} fi