Skip to content

Commit 1781ca7

Browse files
author
wangzaijun
committed
fix
1 parent 550b6cf commit 1781ca7

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

lightllm/server/visualserver/model_infer/model_rpc.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -163,13 +163,9 @@ def _init_env(port, device_id):
163163
# 注册graceful 退出的处理
164164
graceful_registry(inspect.currentframe().f_code.co_name)
165165

166-
auth = lambda sock: (sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1) or (sock, None))
167-
t = ThreadedServer(
168-
VisualModelRpcServer(),
169-
port=port,
170-
protocol_config={"allow_pickle": True},
171-
authenticator=auth,
172-
)
166+
import lightllm.utils.rpyc_fix_utils as _
167+
168+
t = ThreadedServer(VisualModelRpcServer(), port=port, protocol_config={"allow_pickle": True})
173169
t.start()
174170
return
175171

0 commit comments

Comments
 (0)