release v0.7.1

This commit is contained in:
hiyouga
2024-05-16 00:57:16 +08:00
parent 2a67ab3925
commit 1c910079d8
3 changed files with 37 additions and 18 deletions

View File

@@ -5,9 +5,9 @@ from setuptools import find_packages, setup
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", "cli.py"), "r", encoding="utf-8") as f:
file_content = f.read()
pattern = r"{0}\W*=\W*\"([^\"]+)\"".format("__version__")
pattern = r"{}\W*=\W*\"([^\"]+)\"".format("VERSION")
(version,) = re.findall(pattern, file_content)
return version