From b23185b33811d048d13c7341f00455068564a2be Mon Sep 17 00:00:00 2001 From: bojiang <5886138+bojiang@users.noreply.github.com> Date: Wed, 31 Jan 2024 21:42:50 +0800 Subject: [PATCH] fix: worker number of start-http-server (#4462) --- src/_bentoml_impl/server/serving.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/_bentoml_impl/server/serving.py b/src/_bentoml_impl/server/serving.py index 527578cd4c9..468505047d3 100644 --- a/src/_bentoml_impl/server/serving.py +++ b/src/_bentoml_impl/server/serving.py @@ -185,11 +185,11 @@ def serve_http( allocator = ResourceAllocator() if dependency_map is None: dependency_map = {} + if service_name: + svc = svc.find_dependent(service_name) num_workers, worker_envs = allocator.get_worker_env(svc) with tempfile.TemporaryDirectory(prefix="bentoml-uds-") as uds_path: - if service_name: - svc = svc.find_dependent(service_name) - elif not development_mode: + if not service_name and not development_mode: with contextlib.ExitStack() as port_stack: for name, dep_svc in svc.all_services().items(): if name == svc.name: