mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-12 06:27:02 +08:00
fix coordinator for hybrid LB mode
Signed-off-by: Nick Hill <nhill@redhat.com>
This commit is contained in:
parent
60ae223986
commit
36ed9f3467
@ -61,11 +61,12 @@ class DPCoordinator:
|
||||
|
||||
host = parallel_config.data_parallel_master_ip
|
||||
external_lb = parallel_config.data_parallel_external_lb
|
||||
hybrid_lb = parallel_config.data_parallel_hybrid_lb
|
||||
|
||||
# Assume coordinator is colocated with front-end procs when not in
|
||||
# external DP LB mode.
|
||||
# either external or hybrid DP LB mode.
|
||||
front_publish_address = get_engine_client_zmq_addr(
|
||||
local_only=not external_lb, host=host)
|
||||
local_only=not external_lb and not hybrid_lb, host=host)
|
||||
|
||||
local_only_eng = dp_size == parallel_config.data_parallel_size_local
|
||||
back_publish_address = get_engine_client_zmq_addr(local_only_eng, host)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user