update
This commit is contained in:
+2
-2
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user