-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
39 lines (36 loc) · 1.25 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
from setuptools import setup, find_packages
requirements = ["aiohttp",
"evaluate",
"instructor",
"litellm",
"nest_asyncio",
"numpy",
"outlines",
"pandas",
"pydantic",
"PyYAML",
"Requests",
"scikit_learn",
"scipy",
"sentence_transformers",
"termcolor",
"torch",
"tqdm",
"transformers",
"absl-py",
"nltk",
"rouge_score",
"wandb"]
setup(
name="TruthTorchLM", # Your package name
version="0.1.14", # Package version
author="Yavuz Faruk Bakman",
author_email="[email protected]",
description="TruthTorchLM is an open-source library designed to assess truthfulness in language models' outputs. The library integrates state-of-the-art methods, offers comprehensive benchmarking tools across various tasks, and enables seamless integration with popular frameworks like Huggingface and LiteLLM.",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
package_dir={"": "src"}, # Maps the base package directory
packages=find_packages(where="src"), # Automatically find and include all packages
install_requires=requirements, # List of dependencies
python_requires=">=3.10", # Minimum Python version
)