From c34db550a6a3233fa5176f8340e31fffdca05788 Mon Sep 17 00:00:00 2001 From: Aaron <29749331+aarnphm@users.noreply.github.com> Date: Fri, 15 Mar 2024 05:33:18 -0400 Subject: [PATCH] fix(build): explicit set to use alpha version Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com> --- openllm-python/src/_openllm_tiny/_entrypoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openllm-python/src/_openllm_tiny/_entrypoint.py b/openllm-python/src/_openllm_tiny/_entrypoint.py index 0e736c50..ee292db2 100644 --- a/openllm-python/src/_openllm_tiny/_entrypoint.py +++ b/openllm-python/src/_openllm_tiny/_entrypoint.py @@ -283,7 +283,7 @@ def construct_python_options(llm_config, llm_fs): from bentoml._internal.bento.build_config import PythonOptions from openllm.bundle._package import build_editable - packages = ['scipy', 'bentoml[tracing]>=1.2', 'openllm[vllm]'] + packages = ['scipy', 'bentoml[tracing]>=1.2', 'openllm[vllm]>0.4'] if llm_config['requirements'] is not None: packages.extend(llm_config['requirements']) built_wheels = [build_editable(llm_fs.getsyspath('/'), p) for p in ('openllm_core', 'openllm_client', 'openllm')]