-
Notifications
You must be signed in to change notification settings - Fork 28
/
setup.py
36 lines (35 loc) · 880 Bytes
/
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
from setuptools import setup, find_packages
setup(
name='trustllm',
version='0.2.4',
description='TrustLLM',
author='Yue Huang & Siyuan Wu & Haoran Wang',
author_email='trustllm.benchmark@gmail.com',
url='https://github.com/HowieHwong/TrustLLM',
packages=find_packages(),
include_package_data=True,
install_requires=[
'transformers',
'huggingface_hub',
'peft',
'numpy>=1.18.1',
'scipy',
'pandas>=1.0.3',
'scikit-learn',
'openai>=1.0.0',
'tqdm',
'tenacity',
'datasets',
'fschat[model_worker]',
'python-dotenv',
'urllib3',
'anthropic',
'google.generativeai',
'google-api-python-client',
'google.ai.generativelanguage',
'replicate',
'zhipuai>=2.0.1'
],
classifiers=[
],
)