passed img related params to WanTransformerBlock call

Author: prishajain1

src/maxdiffusion/models/wan/transformers/transformer_wan.py

@@ -515,6 +515,8 @@ def init_block(rngs):
           precision=precision,
           attention=attention,
           dropout=dropout,
+          added_kv_proj_dim=added_kv_proj_dim,
+          image_seq_len=pos_embed_seq_len,
       )
 
     self.gradient_checkpoint = GradientCheckpointType.from_str(remat_policy)
@@ -543,6 +545,8 @@ 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