diff --git a/docker/npu.Dockerfile b/docker/npu.Dockerfile index e82bdb1b5..e8922e23f 100644 --- a/docker/npu.Dockerfile +++ b/docker/npu.Dockerfile @@ -75,9 +75,7 @@ ENV LC_ALL=en_US.UTF-8 RUN ${PIP_INSTALL} memfabric-hybrid==1.0.8 ### Install zbal -RUN if [ "$TARGETARCH" = "arm64" ]; then \ - ${PIP_INSTALL} memfabric-zbal==1.1.1; \ - fi +RUN ${PIP_INSTALL} memfabric-zbal==1.2.0 ### Install SGLang Model Gateway RUN ${PIP_INSTALL} sglang-router diff --git a/scripts/ci/npu/npu_ci_install_dependency.sh b/scripts/ci/npu/npu_ci_install_dependency.sh index 99ff530d8..8e2716d85 100755 --- a/scripts/ci/npu/npu_ci_install_dependency.sh +++ b/scripts/ci/npu/npu_ci_install_dependency.sh @@ -54,8 +54,7 @@ PTA_URL="https://gitcode.com/Ascend/pytorch/releases/download/v26.0.0-pytorch2.1 ${PIP_INSTALL} ${PTA_URL} ### Install zbal -${UV_PIP_INSTALL} memfabric-zbal==1.1.1 - +${UV_PIP_INSTALL} memfabric-zbal==1.2.0 ### Install Triton-Ascend ${PIP_INSTALL} triton-ascend==3.2.1.dev20260530 --extra-index-url=https://mirrors.huaweicloud.com/ascend/repos/pypi/nightly --trusted-host triton-ascend.osinfra.cn