coffee
36 lines
976 B
Python
Executable File
36 lines
976 B
Python
Executable File
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",
|
|
<<<<<<< HEAD
|
|
"datasets==3.6.0",
|
|
=======
|
|
<<<<<<< HEAD
|
|
=======
|
|
"datasets==3.6.0",
|
|
>>>>>>> fc9eefc (add pretrain code)
|
|
>>>>>>> 11473a798f338efdf3f70d665a00810099f51174
|
|
"sentencepiece==0.2.0",
|
|
"protobuf==3.20.0",
|
|
"transformers==4.51.3",
|
|
"tokenizers==0.21.1",
|
|
<<<<<<< HEAD
|
|
"setuptools",
|
|
=======
|
|
<<<<<<< HEAD
|
|
=======
|
|
"accelerate==1.7.0",
|
|
>>>>>>> fc9eefc (add pretrain code)
|
|
>>>>>>> 11473a798f338efdf3f70d665a00810099f51174
|
|
],
|
|
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",
|
|
) |