Merge pull request #3930 from MengqingCao/npu

Add Ascend npu doc and dependency Former-commit-id: 7210090e4fc6531b9f6122f104875811a8798185
2024-05-29 18:33:38 +08:00
parent 084be442f2 e2cfd34da0
commit f4e16ae373
3 changed files with 39 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -33,6 +33,7 @@ extra_require = {
    "qwen": ["tiktoken", "transformers_stream_generator"],
    "modelscope": ["modelscope"],
    "quality": ["ruff"],
+    "torch_npu": ["torch==2.1.0", "torch_npu==2.1.0.post3", "decorator"]
 }