Dynamic model class loading (#101)

This commit is contained in:
Cody Yu
2024-01-25 15:29:07 -08:00
committed by GitHub
parent 0147f940dd
commit 3a581e9949
6 changed files with 40 additions and 28 deletions

View File

@@ -20,7 +20,7 @@ dependencies = [
[project.optional-dependencies]
srt = ["aiohttp", "fastapi", "psutil", "rpyc", "torch", "uvloop", "uvicorn",
"zmq", "vllm>=0.2.5", "interegular", "lark", "numba",
"pydantic", "diskcache", "cloudpickle"]
"pydantic", "diskcache", "cloudpickle", "pillow"]
openai = ["openai>=1.0", "numpy"]
anthropic = ["anthropic", "numpy"]
all = ["sglang[srt]", "sglang[openai]", "sglang[anthropic]"]