xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-15 08:54:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Restore preview functionality

Browse Source
This commit is contained in:
kijai 2025-10-23 23:50:09 +03:00
parent 4cbc1a626c
commit f5d53f2a6b
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/ldm/wan/model_multitalk.py
View File

@ -4,6 +4,7 @@ import math
import comfy import comfy
from comfy.ldm.modules.attention import optimized_attention from comfy.ldm.modules.attention import optimized_attention
import logging import logging
import latent_preview
def calculate_x_ref_attn_map(visual_q, ref_k, ref_target_masks): def calculate_x_ref_attn_map(visual_q, ref_k, ref_target_masks):
@ -488,8 +489,13 @@ class InfiniteTalkOuterSampleLoopingWrapper:
# custom previewer callback for full loop progress bar # custom previewer callback for full loop progress bar
pbar = comfy.utils.ProgressBar(total_steps) pbar = comfy.utils.ProgressBar(total_steps)
previewer = latent_preview.get_previewer(model_patcher.load_device, model_patcher.model.latent_format)
def custom_callback(step, x0, x, total_steps): def custom_callback(step, x0, x, total_steps):
pbar.update(1) preview_bytes = None
if previewer:
preview_bytes = previewer.decode_latent_to_preview_image("JPEG", x0)
pbar.update_absolute(pbar.current+1, preview=preview_bytes)
# outer loop start for multiple frame windows # outer loop start for multiple frame windows
for i in range(estimated_iterations): for i in range(estimated_iterations):