423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 13:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update setup.py

Browse Source 
Former-commit-id: 74ff8664d783f428227fb62e6a1313a73cbd337d
This commit is contained in:
hiyouga 2024-03-09 00:14:48 +08:00
parent 398c261c7c
commit 247aab9066
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
setup.py
View File

@ -1,13 +1,14 @@
import os import os
import re import re
from setuptools import setup, find_packages
from setuptools import find_packages, setup
def get_version(): def get_version():
with open(os.path.join("src", "llmtuner", "__init__.py"), "r", encoding="utf-8") as f: with open(os.path.join("src", "llmtuner", "__init__.py"), "r", encoding="utf-8") as f:
file_content = f.read() file_content = f.read()
pattern = r"{0}\W*=\W*\"([^\"]+)\"".format("__version__") pattern = r"{0}\W*=\W*\"([^\"]+)\"".format("__version__")
version, = re.findall(pattern, file_content) (version,) = re.findall(pattern, file_content)
return version return version
@ -29,11 +30,11 @@ extra_require = {
"aqlm": ["aqlm[gpu]>=1.1.0", "transformers @ git+https://github.com/huggingface/transformers.git"], "aqlm": ["aqlm[gpu]>=1.1.0", "transformers @ git+https://github.com/huggingface/transformers.git"],
"galore": ["galore_torch @ git+https://github.com/hiyouga/GaLore.git"], "galore": ["galore_torch @ git+https://github.com/hiyouga/GaLore.git"],
"qwen": ["tiktoken", "transformers_stream_generator"], "qwen": ["tiktoken", "transformers_stream_generator"],
"quality": ["ruff"],
} }
def main(): def main():
setup( setup(
name="llmtuner", name="llmtuner",
version=get_version(), version=get_version(),
@ -51,7 +52,7 @@ def main():
install_requires=get_requires(), install_requires=get_requires(),
extras_require=extra_require, extras_require=extra_require,
classifiers=[ classifiers=[
"Development Status :: 3 - Alpha", "Development Status :: 4 - Beta",
"Intended Audience :: Developers", "Intended Audience :: Developers",
"Intended Audience :: Education", "Intended Audience :: Education",
"Intended Audience :: Science/Research", "Intended Audience :: Science/Research",
@ -61,8 +62,9 @@ def main():
"Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence", "Topic :: Scientific/Engineering :: Artificial Intelligence",
] ],
) )