import os import re from setuptools import setup, find_packages def get_version(): with open(os.path.join("src", "llmtuner", "__init__.py"), "r", encoding="utf-8") as f: file_content = f.read() pattern = r"{0}\W*=\W*\"([^\"]+)\"".format("__version__") version, = re.findall(pattern, file_content) return version def get_requires(): with open("requirements.txt", "r", encoding="utf-8") as f: file_content = f.read() lines = [line.strip() for line in file_content.strip().split("\n") if not line.startswith("#")] return lines def main(): setup( name="llmtuner", version=get_version(), author="hiyouga", author_email="hiyouga" "@" "buaa.edu.cn", description="Easy-to-use fine-tuning framework using PEFT", long_description=open("README.md", "r", encoding="utf-8").read(), long_description_content_type="text/markdown", keywords=["LLaMA", "BLOOM", "Falcon", "LLM", "ChatGPT", "transformer", "pytorch", "deep learning"], license="Apache 2.0 License", url="https://github.com/hiyouga/LLaMA-Efficient-Tuning", package_dir={"": "src"}, packages=find_packages("src"), python_requires=">=3.8.0", install_requires=get_requires(), classifiers=[ "Development Status :: 3 - Alpha", "Intended Audience :: Developers", "Intended Audience :: Education", "Intended Audience :: Science/Research", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent", "Programming Language :: Python :: 3", "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Topic :: Scientific/Engineering :: Artificial Intelligence", ] ) if __name__ == "__main__": main()