From c17a0291731bd484ad911493aa9e869e915c3e7e Mon Sep 17 00:00:00 2001
From: karrix <thomasdice1834793903@gmail.com>
Date: Tue, 22 Oct 2024 17:34:33 +0800
Subject: [PATCH] tweak: back to support all

---
 custom_routes.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/custom_routes.py b/custom_routes.py
index 10ef50f..4eb1563 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -1226,10 +1226,9 @@ async def send_json_override(self, event, data, sid=None):
                 "node_id": node,
                 "node_class": class_type,
             }
-            # if class_type == "PreviewImage":
-            #     logger.info("Skipping preview image")
-            # else:
-            if class_type == "ComfyDeployStdOutputImage":
+            if class_type == "PreviewImage":
+                logger.info("Skipping preview image")
+            else:
                 await update_run_with_output(
                     prompt_id,
                     data.get("output"),