update reqs

InternLM · Oct 18, 2023 · 2054d63 · 2054d63
1 parent bb3cce9
commit 2054d63
Show file tree

Hide file tree

Showing 6 changed files with 28 additions and 18 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,17 +1,3 @@
-accelerate
-datasets
-fastapi
-fire
-gradio
-mmengine
-numpy
-pybind11
-safetensors
-sentencepiece
-setuptools
-shortuuid
-tiktoken
-torch
-transformers>=4.33.0
-tritonclient[all]
-uvicorn
+-r requirements/build.txt
+-r requirements/runtime.txt
+-r requirements/optional.txt
diff --git a/requirements/build.txt b/requirements/build.txt
@@ -0,0 +1,2 @@
+pybind11
+setuptools
diff --git a/requirements/optional.txt b/requirements/optional.txt
@@ -0,0 +1,5 @@
+accelerate
+datasets
+fastapi
+shortuuid
+uvicorn
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
@@ -0,0 +1,9 @@
+fire
+gradio
+mmengine
+numpy
+safetensors
+sentencepiece
+tiktoken
+torch
+transformers>=4.33.0
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -0,0 +1,2 @@
+coverage
+pytest
diff --git a/setup.py b/setup.py
@@ -133,7 +133,13 @@ def gen_packages_items():
               'lmdeploy': lmdeploy_package_data,
           },
           include_package_data=True,
-          install_requires=parse_requirements('requirements.txt'),
+          setup_requires=parse_requirements('requirements/build.txt'),
+          tests_require=parse_requirements('requirements/test.txt'),
+          install_requires=parse_requirements('requirements/runtime.txt'),
+          extras_require={
+              'all': parse_requirements('requirements.txt'),
+              'optional': parse_requirements('requirements/optional.txt'),
+          },
           has_ext_modules=check_ext_modules,
           classifiers=[
               'Programming Language :: Python :: 3.8',