add cli branch in hook

This commit is contained in:
RandomGitUser321 2024-09-20 02:16:52 -04:00 committed by GitHub
parent e7d4782736
commit 63f0910eea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)