derek-thomas's picture
derek-thomas HF staff
Adding setup.py
cd84bc1
raw
history blame
796 Bytes
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",
],
)