add setup file to install dolly_llm

2025-05-14 13:39:27 +08:00
parent 165f61b8b1
commit 2b19c0b1b3
1 changed files with 20 additions and 0 deletions
@@ -0,0 +1,20 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="llm_dolly",  
+    version="0.0.1",      
+    packages=find_packages(include=["src_llm"]),  
+    install_requires=[
+        "torch==2.3.1",
+        "sentencepiece==0.2.0",
+        "protobuf==3.20.0",
+        "transformers==4.51.3",
+        "tokenizers==0.21.1",
+        ],  
+    author="IP127000",  
+    author_email="hanluzhi@outlook.com", 
+    description="LLM-Dolly is a custom LLM built from scratch", 
+    long_description=open('README.md').read(),
+    long_description_content_type="text/markdown", 
+    url="https://github.com/IP127000/LLM-Dolly",
+)