From 0546b01080585b081eaca3ae7bc45000e22be620 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Thu, 15 Feb 2024 23:43:14 -0800 Subject: [PATCH] Tweak RPC parameters --- comfy/distributed/distributed_prompt_queue.py | 3 +-- comfy/distributed/distributed_prompt_worker.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/comfy/distributed/distributed_prompt_queue.py b/comfy/distributed/distributed_prompt_queue.py index ad9a19fde..b4d6a6d2d 100644 --- a/comfy/distributed/distributed_prompt_queue.py +++ b/comfy/distributed/distributed_prompt_queue.py @@ -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 diff --git a/comfy/distributed/distributed_prompt_worker.py b/comfy/distributed/distributed_prompt_worker.py index 9b6e091d5..f82d45f9a 100644 --- a/comfy/distributed/distributed_prompt_worker.py +++ b/comfy/distributed/distributed_prompt_worker.py @@ -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(