File size: 796 Bytes
cd84bc1 |
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 |
from setuptools import setup, find_packages
setup(
name="LLMPerf",
version="0.1.0",
author="Avnish Narayan",
author_email="avnish@anyscale.com",
description="A framework for load testing LLM APIs",
license="Apache-2.0",
packages=find_packages(),
python_requires=">=3.8, <3.11",
install_requires=[
"pydantic<2.5",
"ray",
"pytest>=6.0",
"seaborn>=0.11",
"awscli>=1.22",
"typer>=0.4",
"litellm>=0.1.738",
"num2words",
"transformers",
"tqdm",
"boto3",
"google-cloud-aiplatform"
],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
],
)
|