Merge pull request #8 from merlintang/main

Update the setup file

.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/

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()

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',
+)