mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-04-24 17:12:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

chore: style [skip ci]

Browse Source 
Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
Aaron
2023-06-29 10:54:58 -04:00
parent e52045eda6
commit 7e8ca79c2d
1 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
pyproject.toml
View File

@@ -21,7 +21,18 @@ classifiers = [
"Programming Language :: Python :: Implementation :: PyPy",
]
# NOTE: The below is managed by ./tools/update-optional-dependencies.py
dependencies = ["bentoml[grpc,io]>=1.0.22", "transformers[torch,tokenizers,accelerate]>=4.29.0", "optimum", "attrs>=23.1.0", "cattrs>=23.1.0", "orjson", "inflection", "tabulate[widechars]>=0.9.0", "httpx", "typing_extensions"]
dependencies = [
"bentoml[grpc,io]>=1.0.22",
"transformers[torch,tokenizers,accelerate]>=4.29.0",
"optimum",
"attrs>=23.1.0",
"cattrs>=23.1.0",
"orjson",
"inflection",
"tabulate[widechars]>=0.9.0",
"httpx",
"typing_extensions",
]
description = 'OpenLLM: REST/gRPC API server for running any open Large-Language Model - StableLM, Llama, Alpaca, Dolly, Flan-T5, Custom'
dynamic = ["version"]
keywords = [
@@ -47,15 +58,24 @@ requires-python = ">=3.8"
# NOTE: Don't modify project.optional-dependencies
# as it is managed by ./tools/update-optional-dependencies.py
[project.optional-dependencies]
agents = ["transformers[agents]>=4.30", "diffusers", "soundfile"]
all = [
"openllm[chatglm]",
"openllm[starcoder]",
"openllm[falcon]",
"openllm[mpt]",
"openllm[fine-tune]",
"openllm[openai]",
"openllm[flan-t5]",
"openllm[agents]",
]
chatglm = ["cpm-kernels", "sentencepiece"]
starcoder = ["bitsandbytes"]
falcon = ["einops", "xformers", "safetensors"]
fine-tune = ["peft", "bitsandbytes", "datasets", "accelerate", "deepspeed", "trl"]
flan-t5 = ["flax", "jax", "jaxlib", "tensorflow", "keras"]
mpt = ["triton"]
openai = ["openai", "tiktoken"]
fine-tune = ["peft", "bitsandbytes", "datasets", "accelerate", "deepspeed", "trl"]
agents = ["transformers[agents]>=4.30", "diffusers", "soundfile"]
flan-t5 = ["flax", "jax", "jaxlib", "tensorflow", "keras"]
all = ["openllm[chatglm]", "openllm[starcoder]", "openllm[falcon]", "openllm[mpt]", "openllm[openai]", "openllm[fine-tune]", "openllm[agents]", "openllm[flan-t5]"]
starcoder = ["bitsandbytes"]
[project.urls]
Documentation = "https://github.com/bentoml/openllm#readme"