wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-12 15:20:51 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Tweak RPC parameters

Browse Source
This commit is contained in:
doctorpangloss 2024-02-15 23:43:14 -08:00
parent a62f2c00ed
commit 0546b01080
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/distributed/distributed_prompt_queue.py
View File

@ -273,8 +273,7 @@ class DistributedPromptQueue(AbstractPromptQueue):
return
self._connection = await connect_robust(self._connection_uri, loop=self._loop)
self._channel = await self._connection.channel()
self._rpc = await JsonRPC.create(channel=self._channel, auto_delete=True, durable=False, exclusive=True)
self._rpc.host_exceptions = True
self._rpc = await JsonRPC.create(channel=self._channel, auto_delete=True, durable=False)
if self._is_caller:
self._caller_progress_handlers = ProgressHandlers(self._rpc, self._caller_server, self._queue_name)
# this makes the queue available to complete work items

3
comfy/distributed/distributed_prompt_worker.py
View File

@ -64,8 +64,7 @@ class DistributedPromptWorker:
logging.error(f"failed to connect to self._connection_uri={self._connection_uri}", connection_error)
raise connection_error
self._channel = await self._connection.channel()
self._rpc = await JsonRPC.create(channel=self._channel, auto_delete=True, durable=False, exclusive=True)
self._rpc.host_exceptions = True
self._rpc = await JsonRPC.create(channel=self._channel, auto_delete=True, durable=False)
if self._embedded_comfy_client is None:
self._embedded_comfy_client = EmbeddedComfyClient(