passed img related params to WanTransformerBlock call

prishajain1 · prishajain1 · commit 698a5d0b23f0 · 2025-12-17T14:15:40.000+05:30
diff --git a/src/maxdiffusion/models/wan/transformers/transformer_wan.py b/src/maxdiffusion/models/wan/transformers/transformer_wan.py
@@ -545,8 +545,6 @@ def init_block(rngs):
             precision=precision,
             attention=attention,
             enable_jax_named_scopes=enable_jax_named_scopes,
-            added_kv_proj_dim=added_kv_proj_dim,
-            image_seq_len=pos_embed_seq_len,
         )
         blocks.append(block)
       self.blocks = blocks

Original file line number	Diff line number	Diff line change
`@@ -545,8 +545,6 @@ def init_block(rngs):`
`545`	`545`	`precision=precision,`
`546`	`546`	`attention=attention,`
`547`	`547`	`enable_jax_named_scopes=enable_jax_named_scopes,`
`548`		`- added_kv_proj_dim=added_kv_proj_dim,`
`549`		`- image_seq_len=pos_embed_seq_len,`
`550`	`548`	`)`
`551`	`549`	`blocks.append(block)`
`552`	`550`	`self.blocks = blocks`