Fix missing numpy dependency in pyproject.toml (#524)

This commit is contained in:
Fabian Preiß
2024-06-10 21:13:50 +02:00
committed by GitHub
parent f6dbd24043
commit 542bc733d6

View File

@@ -16,13 +16,14 @@ classifiers = [
dependencies = [
"requests",
"tqdm",
"numpy",
]
[project.optional-dependencies]
srt = ["aiohttp", "fastapi", "psutil", "rpyc", "torch", "uvloop", "uvicorn",
"zmq", "vllm==0.4.3", "interegular", "pydantic", "pillow", "packaging", "huggingface_hub", "hf_transfer", "outlines>=0.0.34"]
openai = ["openai>=1.0", "numpy", "tiktoken"]
anthropic = ["anthropic>=0.20.0", "numpy"]
openai = ["openai>=1.0", "tiktoken"]
anthropic = ["anthropic>=0.20.0"]
litellm = ["litellm>=1.0.0"]
all = ["sglang[srt]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]