Add Ascend installation adapter (#2817)

Author: zhabuye

requirements/runtime_ascend.txt

@@ -0,0 +1,22 @@
+accelerate>=0.29.3
+dlinfer-ascend
+einops
+fastapi
+fire
+mmengine-lite
+numpy<2.0.0
+openai
+outlines<0.1.0
+peft<=0.11.1
+pillow
+protobuf
+pydantic>2.0.0
+pynvml
+safetensors
+sentencepiece
+shortuuid
+tiktoken
+torch<=2.4.0,>=2.0.0
+torchvision<=0.19.0,>=0.15.0
+transformers
+uvicorn

requirements_ascend.txt

@@ -0,0 +1,4 @@
+-r requirements/build.txt
+-r requirements/runtime_ascend.txt
+-r requirements/lite.txt
+-r requirements/serve.txt

setup.py

@@ -4,6 +4,14 @@
 
 from setuptools import find_packages, setup
 
+npu_available = False
+try:
+    import torch_npu
+
+    npu_available = torch_npu.npu.is_available()
+except ImportError:
+    pass
+
 pwd = os.path.dirname(__file__)
 version_file = 'lmdeploy/version.py'
 
@@ -145,11 +153,17 @@ def gen_packages_items():
         include_package_data=True,
         setup_requires=parse_requirements('requirements/build.txt'),
         tests_require=parse_requirements('requirements/test.txt'),
-        install_requires=parse_requirements('requirements/runtime.txt'),
+        install_requires=parse_requirements(
+            'requirements/runtime_ascend.txt'
+            if npu_available else 'requirements/runtime.txt'),
         extras_require={
-            'all': parse_requirements('requirements.txt'),
-            'lite': parse_requirements('requirements/lite.txt'),
-            'serve': parse_requirements('requirements/serve.txt')
+            'all':
+            parse_requirements('requirements_ascend.txt'
+                               if npu_available else 'requirements.txt'),
+            'lite':
+            parse_requirements('requirements/lite.txt'),
+            'serve':
+            parse_requirements('requirements/serve.txt')
         },
         has_ext_modules=check_ext_modules,
         classifiers=[