Ignore more network related errors during websocket communication. (#…

…5269) Intermittent network issues during websocket communication should not crash ComfyUi process. Co-authored-by: Xiaodong Xie <[email protected]>
Comfy-Org · Oct 17, 2024 · 89f1589 · 89f1589
1 parent 6715899
commit 89f1589
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():