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/assets/wechat.jpg b/assets/wechat.jpg index 099308d9..6e692cde 100644 Binary files a/assets/wechat.jpg and b/assets/wechat.jpg differ diff --git a/assets/wechat_npu.jpg b/assets/wechat_npu.jpg index 3e4c5798..64eb1888 100644 Binary files a/assets/wechat_npu.jpg and b/assets/wechat_npu.jpg differ 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"],