tweak: back to support all

2024-10-22 17:34:33 +08:00 · 2024-10-22 17:34:33 +08:00 · c17a029173
commit c17a029173
parent c82f0c2bf0
1 changed files with 3 additions and 4 deletions
--- 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"),