update torch-npu version

Former-commit-id: a70d7fcf2967eb30280a1fb845b39db7878f535c
2024-05-29 10:05:11 +00:00
parent 668dea9706
commit e2cfd34da0
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,7 @@ extra_require = {
    "qwen": ["tiktoken", "transformers_stream_generator"],
    "modelscope": ["modelscope"],
    "quality": ["ruff"],
-    "torch_npu": ["torch==2.1.0", "torch_npu==2.1.0", "decorator"]
+    "torch_npu": ["torch==2.1.0", "torch_npu==2.1.0.post3", "decorator"]
 }