Fix dimension mismatch in wan i2v high/low noise branches caused by redundant transpose

Author: Perseus14

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p2.py

@@ -699,7 +699,6 @@ def high_noise_branch(operands):
         mask_high,
         _,
     ) = operands
-    latents_input = jnp.transpose(latents_input, (0, 4, 1, 2, 3))
     noise_pred, latents_out = transformer_forward_pass(
         high_noise_graphdef,
         high_noise_state,
@@ -728,7 +727,6 @@ def low_noise_branch(operands):
         _,
         mask_low,
     ) = operands
-    latents_input = jnp.transpose(latents_input, (0, 4, 1, 2, 3))
     noise_pred, latents_out = transformer_forward_pass(
         low_noise_graphdef,
         low_noise_state,