-
Notifications
You must be signed in to change notification settings - Fork 20
/
setup.py
47 lines (45 loc) · 1.42 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
39
40
41
42
43
44
45
46
47
from setuptools import setup, find_packages
with open('README.md', 'r') as fh:
long_description = fh.read()
setup(
name='llm_benchmark',
version='0.3.22',
author='Jason Chuang',
author_email='[email protected]',
description='LLM Benchmark',
long_description=long_description,
long_description_content_type='text/markdown',
url='https://github.com/aidatatools/ollama-benchmark',
packages=find_packages(),
package_data={'llm_benchmark': ['data/*.yml', 'data/img/*.jpg']},
classifiers=[
'Programming Language :: Python :: 3',
'License :: OSI Approved :: MIT License',
'Operating System :: OS Independent',
],
python_requires='>=3.9',
install_requires=[
'click>=8.0.0',
'typer==0.11.0',
'ollama==0.2.1',
'pyyaml==6.0.1',
'requests==2.31.0',
'psutil==5.9.8',
'GPUtil==1.4.0',
'lib-platform==1.2.10',
'setuptools==69.1.0'
],
entry_points={
'console_scripts': [
'llm_benchmark = llm_benchmark.main:app',
],
},
# This line enables editable installs
# With 'pip install -e .' equivalent
# to install your package in editable mode
# so changes in your source code are immediately reflected
# without needing to reinstall
options={'bdist_wheel': {'universal': True}},
setup_requires=['setuptools>=69.1.0', 'wheel'],
editable=True
)