Skip to content

Commit 6757bd2

Browse files
committed
update if
Signed-off-by: SlightwindSec <[email protected]>
1 parent ceadc27 commit 6757bd2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm_ascend/distributed/mooncake/mooncake_store_connector_v1.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(self, vllm_config: VllmConfig, role: KVConnectorRole):
4545
)
4646

4747
assert self.connector_worker is not None
48-
if vllm_config.parallel_config.rank == 0:
48+
if vllm_config.parallel_config.rank == 0 and self.kv_role == "kv_producer":
4949
self.lookup_server = MooncakeLookupServer(
5050
self.connector_worker, vllm_config, self.use_layerwise)
5151

@@ -160,9 +160,10 @@ def get_zmq_rpc_path_mooncake(
160160
class MooncakeStoreConnectorV1Scheduler:
161161

162162
def __init__(self, vllm_config: "VllmConfig", use_layerwise):
163-
self.client = MooncakeLookupClient(vllm_config)
164163
self.use_layerwise = use_layerwise
165164
self.kv_role = vllm_config.kv_transfer_config.kv_role
165+
self.client = MooncakeLookupClient(
166+
vllm_config) if self.kv_role == "kv_producer" else None
166167
self.consumer_is_to_load = vllm_config.kv_transfer_config.kv_connector_extra_config.get(
167168
"consumer_is_to_load", False)
168169
self.load_async = vllm_config.kv_transfer_config.kv_connector_extra_config.get(

0 commit comments

Comments
 (0)