From cbb65567a9053326d45ae4f4e5242c18b04daf0f Mon Sep 17 00:00:00 2001 From: Yaowei Zheng Date: Tue, 1 Jul 2025 21:23:48 +0800 Subject: [PATCH] [assets] update wechat (#8517) --- .github/workflows/label_issue.yml | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/label_issue.yml b/.github/workflows/label_issue.yml index b7469f6a..3d0424c7 100644 --- a/.github/workflows/label_issue.yml +++ b/.github/workflows/label_issue.yml @@ -19,7 +19,7 @@ jobs: ISSUE_TITLE: ${{ github.event.issue.title }} run: | LABEL="" - NPU_KEYWORDS=(npu huawei ascend 华为 昇腾) + NPU_KEYWORDS=(npu huawei ascend 华为 昇腾 910) ISSUE_TITLE_LOWER=$(echo $ISSUE_TITLE | tr '[:upper:]' '[:lower:]') for KEYWORD in ${NPU_KEYWORDS[@]}; do if [[ $ISSUE_TITLE_LOWER == *$KEYWORD* ]] && [[ $ISSUE_TITLE_LOWER != *input* ]]; then diff --git a/setup.py b/setup.py index 11235be7..3c5b445b 100644 --- a/setup.py +++ b/setup.py @@ -43,7 +43,7 @@ def get_console_scripts() -> list[str]: extra_require = { "torch": ["torch>=2.0.0", "torchvision>=0.15.0"], - "torch-npu": ["torch==2.4.0", "torch-npu==2.4.0.post2", "decorator"], + "torch-npu": ["torch==2.5.1", "torchvision==0.20.1", "torch-npu==2.5.1", "decorator"], "metrics": ["nltk", "jieba", "rouge-chinese"], "deepspeed": ["deepspeed>=0.10.0,<=0.16.9"], "liger-kernel": ["liger-kernel>=0.5.5"],