From de279dd0605d418797555744b1969845e71bc8f0 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 2 Jul 2025 17:15:18 +0300 Subject: [PATCH] update --- latent_preview.py | 4 ++-- nodes.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/latent_preview.py b/latent_preview.py index 36a253e..e4144de 100644 --- a/latent_preview.py +++ b/latent_preview.py @@ -98,9 +98,9 @@ def prepare_callback(model, steps, x0_output_dict=None): preview_format = "JPEG" previewer = get_previewer(model.load_device, model.model.latent_format) - print("previewer: ", previewer) - pbar = comfy.utils.ProgressBar(steps) + if steps is not None: + pbar = comfy.utils.ProgressBar(steps) def callback(step, x0, x, total_steps): if x0_output_dict is not None: x0_output_dict["x0"] = x0 diff --git a/nodes.py b/nodes.py index 16b166b..a271c5a 100644 --- a/nodes.py +++ b/nodes.py @@ -3838,6 +3838,7 @@ class WanVideoSampler: estimated_iterations = max(1, min(max_frames_num, len(multitalk_audio_embedding)) // (frame_num - motion_frame) + 1) loop_pbar = tqdm(total=estimated_iterations, desc="Generating video clips") comfy_pbar = ProgressBar(estimated_iterations) + callback = prepare_callback(patcher, None) iteration_count = 0 audio_embedding = [multitalk_audio_embedding] @@ -3958,7 +3959,7 @@ class WanVideoSampler: cfg[idx], text_embeds["prompt_embeds"], text_embeds["negative_prompt_embeds"], - timestep, idx, y.squeeze(0), clip_embeds, control_latents, vace_data, unianim_data, audio_proj, control_camera_latents, add_cond, + timestep, idx, y.squeeze(0), clip_embeds.to(dtype), control_latents, vace_data, unianim_data, audio_proj, control_camera_latents, add_cond, cache_state=self.cache_state, multitalk_audio_embeds=audio_embs) if callback is not None: