|
- import os
- import re
- from setuptools import setup, find_packages
-
-
- def get_version():
- with open(os.path.join("src", "glmtuner", "__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="glmtuner",
- version=get_version(),
- author="hiyouga",
- author_email="hiyouga" "@" "buaa.edu.cn",
- description="Fine-tuning ChatGLM-6B with PEFT",
- long_description=open("README.md", "r", encoding="utf-8").read(),
- long_description_content_type="text/markdown",
- keywords=["ChatGLM", "LLM", "ChatGPT", "transformer", "pytorch", "deep learning"],
- license="Apache 2.0 License",
- url="https://github.com/hiyouga/ChatGLM-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()
|