wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 20:30:25 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Fix linting issue

Browse Source
This commit is contained in:
doctorpangloss 2025-09-03 15:08:16 -07:00
parent 9ac2864408
commit 6cffd4c4c2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ldm/wan/model.py
View File

@ -1252,6 +1252,7 @@ class WanModel_S2V(WanModel):
num_embeds = x.shape[-3] * 4 num_embeds = x.shape[-3] * 4
audio_emb_global, audio_emb = self.casual_audio_encoder(audio_embed[:, :, :, :num_embeds]) audio_emb_global, audio_emb = self.casual_audio_encoder(audio_embed[:, :, :, :num_embeds])
else: else:
audio_emb_global = None
audio_emb = None audio_emb = None
# embeddings # embeddings
@ -1311,7 +1312,7 @@ class WanModel_S2V(WanModel):
x = out["img"] x = out["img"]
else: else:
x = block(x, e=e0, freqs=freqs, context=context) x = block(x, e=e0, freqs=freqs, context=context)
if audio_emb is not None: if audio_emb is not None and audio_emb_global is not None:
x = self.audio_injector(x, i, audio_emb, audio_emb_global, seq_len) x = self.audio_injector(x, i, audio_emb, audio_emb_global, seq_len)
# head # head
x = self.head(x, e) x = self.head(x, e)