From b8022cc6b8ae84c09ebd0a8512aedc5301ce9716 Mon Sep 17 00:00:00 2001 From: coffee Date: Wed, 14 May 2025 13:39:27 +0800 Subject: [PATCH] add setup file to install dolly_llm --- setup.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100755 setup.py diff --git a/setup.py b/setup.py new file mode 100755 index 0000000..492bd45 --- /dev/null +++ b/setup.py @@ -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", +) \ No newline at end of file