File size: 1,671 Bytes
3a464db | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | 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",
)
# optionals
# lm-eval==0.3.0 |