From 3b8d2385b787eaab972c373e363763db2651ab3b Mon Sep 17 00:00:00 2001 From: Xiaodong Xie Date: Thu, 17 Oct 2024 18:25:42 +0200 Subject: [PATCH] Ignore more network related errors during websocket communication. Intermittent network issues during websocket communication should not crash ComfyUi process. --- server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server.py b/server.py index c7bf6622d..ada6d90c3 100644 --- a/server.py +++ b/server.py @@ -40,7 +40,7 @@ class BinaryEventTypes: async def send_socket_catch_exception(function, message): try: await function(message) - except (aiohttp.ClientError, aiohttp.ClientPayloadError, ConnectionResetError) as err: + except (aiohttp.ClientError, aiohttp.ClientPayloadError, ConnectionResetError, BrokenPipeError, ConnectionError) as err: logging.warning("send error: {}".format(err)) def get_comfyui_version():