From b19d33b9d205911ffd191bd0eadddd162b692e68 Mon Sep 17 00:00:00 2001
From: MengqingCao <cmq0113@163.com>
Date: Wed, 29 May 2024 10:05:11 +0000
Subject: [PATCH] update torch-npu version

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index f72eefb6..06e6b750 100644
--- 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"]
 }