From 3addf9c56fbe235931e08d38a5fce0fc42551ab8 Mon Sep 17 00:00:00 2001 From: mingjie tang Date: Mon, 28 Aug 2023 15:23:52 -0700 Subject: [PATCH 1/3] update the git ignore --- .gitignore | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index f52bf300..df171882 100644 --- a/.gitignore +++ b/.gitignore @@ -157,7 +157,7 @@ cython_debug/ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. -#.idea/ +.idea/ # IDEs .vscode/ From 6cff9ce0eb8e81e64234bd1e4abbcbd63066f06a Mon Sep 17 00:00:00 2001 From: mingjie tang Date: Mon, 28 Aug 2023 15:30:44 -0700 Subject: [PATCH 2/3] update the setup file --- setup.py | 27 +++++++++++++++++++++++---- 1 file changed, 23 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 6d51f744..5b75d507 100644 --- a/setup.py +++ b/setup.py @@ -1,5 +1,24 @@ -from setuptools import setup, find_packages +import os -setup(name='aspen', - version='0.2', - packages=find_packages(exclude=["test"])) +import pkg_resources +from setuptools import find_packages, setup + +setup( + name='llm-multi-lora-finetune', + version='0.0.1', + packages=find_packages(), + description='A tool for fine-tuning large language models using LoRA or QLoRA', + classifiers=[ + "Programming Language :: Python :: 3", + "License :: OSI Approved :: Apache License", + "Operating System :: OS Independent", + ], + install_requires=[ + str(r) + for r in pkg_resources.parse_requirements( + open(os.path.join(os.path.dirname(__file__), "requirements.txt")) + ) + ], + include_package_data=True, + python_requires='>=3.6', +) \ No newline at end of file From 03735494dc1c7703ffaebc1546c85bb22486c57d Mon Sep 17 00:00:00 2001 From: mingjie tang Date: Mon, 28 Aug 2023 15:37:07 -0700 Subject: [PATCH 3/3] update the test cases --- aspen/tests/loader_test.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 aspen/tests/loader_test.py diff --git a/aspen/tests/loader_test.py b/aspen/tests/loader_test.py new file mode 100644 index 00000000..ef6f7c88 --- /dev/null +++ b/aspen/tests/loader_test.py @@ -0,0 +1,21 @@ +import unittest + + +class TestDataLoader(unittest.TestCase): + def test_load_llma(self): + """ + Todo: add the test function here + :return: + """ + self.assertEqual(1 + 1, 2) + + def test_load_llam2(self): + """ + Todo: add the test function here + :return: + """ + self.assertEqual(2 - 1, 1) + + +if __name__ == '__main__': + unittest.main() \ No newline at end of file