xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-11 03:46:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

add cli branch in hook

Browse Source
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
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -170,7 +170,10 @@ def hijack_progress(server):
server.send_sync("progress", progress, server.client_id) server.send_sync("progress", progress, server.client_id)
if preview_image is not None: 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) comfy.utils.set_progress_bar_global_hook(hook)