-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
38 lines (35 loc) · 1.07 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import setuptools
import io
project_name = "jllm"
version = "4.0.3"
setuptools.setup(
name=project_name,
version=version,
author="Jian Lu",
license="Apache 2.0",
description=("Running Large Language Model easily, faster and low-cost."),
url="https://github.com/janelu9/EasyLLM",
project_urls={
"Homepage": "https://github.com/janelu9/EasyLLM",
},
long_description=io.open("README.md", "r", encoding="utf-8").read(),
long_description_content_type="text/markdown",
classifiers=[
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"License :: OSI Approved :: Apache Software License",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
packages=setuptools.find_packages(),
include_package_data=True,
python_requires='>=3.9',
install_requires=[
"deepspeed",
"protobuf",
"sentencepiece",
"transformers",
"pyarrow",
"tiktoken"
],
)