From 63f0910eea37afa7f497996326fea8c303036a20 Mon Sep 17 00:00:00 2001 From: RandomGitUser321 Date: Fri, 20 Sep 2024 02:16:52 -0400 Subject: [PATCH] add cli branch in hook --- main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index f5b633e5b..ae61177c3 100644 --- a/main.py +++ b/main.py @@ -170,7 +170,10 @@ def hijack_progress(server): server.send_sync("progress", progress, server.client_id) if preview_image is not None: - server.send_sync(BinaryEventTypes.UNENCODED_PREVIEW_IMAGE, preview_image, server.client_id) + if args.latent_preview_api: + server.send_sync("latent_preview", preview_image, server.client_id) + else: + server.send_sync(BinaryEventTypes.UNENCODED_PREVIEW_IMAGE, preview_image, server.client_id) comfy.utils.set_progress_bar_global_hook(hook)