Skip to content

Commit 7f4620f

Browse files
author
wangzaijun
committed
fix result_queue
1 parent 3ec84db commit 7f4620f

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

lightllm/server/router/model_infer/model_rpc.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,14 +45,12 @@ def __init__(
4545
rpc_event: multiprocessing.Event,
4646
rpc_finished_event: multiprocessing.Event,
4747
info_queue: mp.Queue,
48-
result_queue: mp.Queue,
4948
mem_queue: mp.Queue,
5049
):
5150
super().__init__()
5251
self.args: StartArgs = args
5352
self.node_world_size = node_world_size
5453
self.info_queue = info_queue
55-
self.result_queue = result_queue
5654
self.mem_queue = mem_queue
5755
self.rpc_event = rpc_event
5856
self.rpc_finished_event = rpc_finished_event
@@ -218,7 +216,6 @@ def _init_env(
218216
rank_in_node,
219217
node_world_size,
220218
info_queue,
221-
result_queue,
222219
mem_queue,
223220
router_lock,
224221
rpc_event: mp.Event,
@@ -237,7 +234,7 @@ def _init_env(
237234
g_router_lock.obj = router_lock
238235

239236
model_rpc_server = ModelRpcServer(
240-
args, rank, rank_in_node, node_world_size, rpc_event, rpc_finished_event, info_queue, result_queue, mem_queue
237+
args, rank, rank_in_node, node_world_size, rpc_event, rpc_finished_event, info_queue, mem_queue
241238
)
242239
success_event.set()
243240

@@ -253,7 +250,6 @@ async def start_model_process(
253250
rpc_event,
254251
rpc_finished_event,
255252
info_queue: mp.Queue,
256-
result_queue: mp.Queue,
257253
mem_queue: mp.Queue,
258254
router_lock: mp.Queue,
259255
):
@@ -268,7 +264,6 @@ async def start_model_process(
268264
rank_in_node,
269265
node_world_size,
270266
info_queue,
271-
result_queue,
272267
mem_queue,
273268
router_lock,
274269
rpc_event,

0 commit comments

Comments
 (0)