revert unnecessary

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>

Author: wangxiyuan

pyproject.toml

@@ -12,7 +12,8 @@ requires = [
     "scipy",
     "setuptools>=64",
     "setuptools-scm>=8",
-    "torch==2.7.1",
+    "torch-npu==2.7.1.dev20250724",
+    "torch>=2.7.1",
     "torchvision",
     "wheel",
     "msgpack",

requirements.txt

@@ -10,7 +10,7 @@ pyyaml
 scipy
 setuptools>=64
 setuptools-scm>=8
-torch==2.7.1
+torch>=2.7.1
 torchvision
 wheel
 
@@ -21,3 +21,7 @@ quart
 # Required for N-gram speculative decoding
 numba
 
+# Install torch_npu
+--pre
+--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi
+torch-npu==2.7.1.dev20250724

setup.py

@@ -202,9 +202,9 @@ def configure(self, ext: CMakeExtension) -> None:
 
         torch_npu_command = "python3 -m pip show torch-npu | grep '^Location:' | awk '{print $2}'"
         try:
-            # torch_npu_path = subprocess.check_output(
-            #     torch_npu_command, shell=True).decode().strip()
-            torch_npu_path = "/usr/local/lib/python3.11/site-packages/torch_npu"
+            torch_npu_path = subprocess.check_output(
+                torch_npu_command, shell=True).decode().strip()
+            torch_npu_path += "/torch_npu"
         except subprocess.CalledProcessError as e:
             raise RuntimeError(f"Retrieve torch version version failed: {e}")