@@ -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