Skip to content

Commit d3bafb2

Browse files
committed
Revert unneeded changes
1 parent e4a0d34 commit d3bafb2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

comfy/ldm/wan/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1510,7 +1510,7 @@ def __init__(self,
15101510
operations=None,
15111511
):
15121512

1513-
super().__init__(model_type='i2v', patch_size=patch_size, text_len=text_len, in_dim=in_dim, dim=dim, ffn_dim=ffn_dim, freq_dim=freq_dim, text_dim=text_dim, out_dim=out_dim, num_heads=num_heads, num_layers=num_layers, window_size=window_size, qk_norm=qk_norm, cross_attn_norm=cross_attn_norm, eps=eps, flf_pos_embed_token_number=flf_pos_embed_token_number, wan_attn_block_class=WanAttentionBlockAudio, image_model=image_model, device=device, dtype=dtype, operations=operations)
1513+
super().__init__(model_type='t2v', patch_size=patch_size, text_len=text_len, in_dim=in_dim, dim=dim, ffn_dim=ffn_dim, freq_dim=freq_dim, text_dim=text_dim, out_dim=out_dim, num_heads=num_heads, num_layers=num_layers, window_size=window_size, qk_norm=qk_norm, cross_attn_norm=cross_attn_norm, eps=eps, flf_pos_embed_token_number=flf_pos_embed_token_number, wan_attn_block_class=WanAttentionBlockAudio, image_model=image_model, device=device, dtype=dtype, operations=operations)
15141514

15151515
self.audio_proj = AudioProjModel(seq_len=8, blocks=5, channels=1280, intermediate_dim=512, output_dim=1536, context_tokens=audio_token_num, dtype=dtype, device=device, operations=operations)
15161516

@@ -1548,7 +1548,7 @@ def forward_orig(
15481548

15491549
# context
15501550
context = self.text_embedding(context)
1551-
context_img_len = 0
1551+
context_img_len = None
15521552

15531553
if audio_embed is not None:
15541554
if reference_latent is not None:

0 commit comments

Comments
 (0)