mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-05-19 14:16:22 -04:00
chore: update deps for transformers and vllm
Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
@@ -39,7 +39,7 @@ classifiers = [
|
||||
]
|
||||
dependencies = [
|
||||
"bentoml[io]>=1.1.2",
|
||||
"transformers[torch,tokenizers,accelerate]>=4.32.1",
|
||||
"transformers[torch,tokenizers]>=4.34.0",
|
||||
"openllm-client",
|
||||
"openllm-core",
|
||||
"safetensors",
|
||||
@@ -115,7 +115,7 @@ openai = ["openai[embeddings]", "tiktoken"]
|
||||
opt = ["flax>=0.7", "jax", "jaxlib", "tensorflow", "keras"]
|
||||
playground = ["jupyter", "notebook", "ipython", "jupytext", "nbformat"]
|
||||
starcoder = ["bitsandbytes"]
|
||||
vllm = ["vllm>=0.1.7", "ray"]
|
||||
vllm = ["vllm>=0.2.0", "ray"]
|
||||
|
||||
[tool.hatch.version]
|
||||
fallback-version = "0.0.0"
|
||||
|
||||
Reference in New Issue
Block a user