This commit is contained in:
kijai
2025-07-02 17:15:18 +03:00
parent ebe3068a8d
commit de279dd060
2 changed files with 4 additions and 3 deletions
+2 -2
View File
@@ -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
+2 -1
View File
@@ -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: