Skip to content

Commit 1c184c2

Browse files
Fix issue with s2v node when extending past audio length. (Comfy-Org#9608)
1 parent edde0b5 commit 1c184c2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

comfy_extras/nodes_wan.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -893,9 +893,10 @@ def wan_sound_to_video(positive, negative, vae, width, height, length, batch_siz
893893
audio_embed_bucket = audio_embed_bucket.permute(0, 2, 3, 1)
894894

895895
audio_embed_bucket = audio_embed_bucket[:, :, :, frame_offset:frame_offset + batch_frames]
896-
positive = node_helpers.conditioning_set_values(positive, {"audio_embed": audio_embed_bucket})
897-
negative = node_helpers.conditioning_set_values(negative, {"audio_embed": audio_embed_bucket * 0.0})
898-
frame_offset += batch_frames
896+
if audio_embed_bucket.shape[3] > 0:
897+
positive = node_helpers.conditioning_set_values(positive, {"audio_embed": audio_embed_bucket})
898+
negative = node_helpers.conditioning_set_values(negative, {"audio_embed": audio_embed_bucket * 0.0})
899+
frame_offset += batch_frames
899900

900901
if ref_image is not None:
901902
ref_image = comfy.utils.common_upscale(ref_image[:1].movedim(-1, 1), width, height, "bilinear", "center").movedim(1, -1)

0 commit comments

Comments
 (0)