| import os |
| from setuptools import find_packages |
| from setuptools import setup |
|
|
| folder = os.path.dirname(__file__) |
| _third_party = os.path.normpath(os.path.join(folder, "..", "third_party")) |
| version_path = os.path.join(_third_party, "lmflow", "version.py") |
|
|
| __version__ = None |
| with open(version_path) as f: |
| exec(f.read(), globals()) |
|
|
| req_path = os.path.join(folder, "requirements.txt") |
| install_requires = [] |
| if os.path.exists(req_path): |
| with open(req_path) as fp: |
| install_requires = [line.strip() for line in fp] |
|
|
| extra_require = { |
| "multimodal": ["Pillow"], |
| "vllm": ["vllm>=0.4.3"], |
| "ray": ["ray>=2.22.0"], |
| "gradio": ["gradio"], |
| "flask": ["flask", "flask_cors"], |
| "flash_attn": ["flash-attn>=2.0.2"], |
| "trl": ["trl==0.8.0"] |
| } |
|
|
| readme_path = os.path.join(folder, "README.md") |
| readme_contents = "" |
| if os.path.exists(readme_path): |
| with open(readme_path, encoding="utf-8") as fp: |
| readme_contents = fp.read().strip() |
|
|
| setup( |
| name="lmflow", |
| version=__version__, |
| description="LMFlow: Large Model Flow.", |
| author="The LMFlow Team", |
| long_description=readme_contents, |
| long_description_content_type="text/markdown", |
| package_dir={"": _third_party}, |
| packages=find_packages(where=_third_party), |
| package_data={}, |
| install_requires=install_requires, |
| extras_require=extra_require, |
| classifiers=[ |
| "Intended Audience :: Science/Research/Engineering", |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| "Programming Language :: Python :: 3.9", |
| "Programming Language :: Python :: 3.10", |
| ], |
| requires_python=">=3.9", |
| ) |
|
|
| |
| |
