add Ascend npu doc and dependency

Former-commit-id: 803d9f142a294f8c1e0b4e2046c214b0857ccfd6
2024-05-28 01:33:54 +00:00
parent 26f293d587
commit e45a9d70fc
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.2.0", "torch_npu==2.2.0", "decorator"]
 }