Don't display JPEG images from websocket (used for ksampler preview) #…

…1330
Acly · Oct 28, 2024 · 0d9e938 · 0d9e938
1 parent 2354d4b
commit 0d9e938
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ai_diffusion/comfy_client.py b/ai_diffusion/comfy_client.py
@@ -721,9 +721,7 @@ def _extract_message_png_image(data: memoryview):
     if len(data) > s:
         event, format = struct.unpack_from(">II", data)
         # ComfyUI server.py: BinaryEventTypes.PREVIEW_IMAGE=1
-        if event == 1:
-            if not (format == 1 or format == 2):  # JPEG=1, PNG=2
-                log.warning(f"Websocket binary data has unknwon image format '{format}'")
+        if event == 1 and format == 2:  # format: JPEG=1, PNG=2
             return Image.from_bytes(data[s:])
     return None