Skip to content

Commit 2035c5d

Browse files
hiworldwzjwangzaijun
andauthored
fix rpyc tcp delay (#646)
Co-authored-by: wangzaijun <wangzaijun@sensetime.com>
1 parent a1d0e47 commit 2035c5d

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

lightllm/server/router/model_infer/mode_backend/continues_batch/decode_node_impl/decode_kv_move_manager.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,8 @@ def timer_loop(self):
281281

282282

283283
def _init_env(args, info_queue: mp.Queue, mem_queues: List[mp.Queue], event: mp.Event):
284+
import lightllm.utils.rpyc_fix_utils as _
285+
284286
# 注册graceful 退出的处理
285287
from lightllm.utils.graceful_utils import graceful_registry
286288
import inspect

lightllm/server/router/model_infer/mode_backend/continues_batch/prefill_node_impl/prefill_kv_move_manager.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ async def wait_all_future_finish(self, futures: List[AsyncResult]):
202202

203203

204204
def _init_env(args, info_queue: mp.Queue, mem_queues: List[mp.Queue], event: mp.Event):
205+
import lightllm.utils.rpyc_fix_utils as _
206+
205207
# 注册graceful 退出的处理
206208
from lightllm.utils.graceful_utils import graceful_registry
207209
import inspect

0 commit comments

Comments
 (0)