removed nornalisation transpose

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline.py

@@ -532,8 +532,6 @@ def prepare_latents_i2v_base(
 
       with self.mesh, nn_partitioning.axis_rules(self.config.logical_axis_rules):
           encoded_output = self.vae.encode(video_condition)[0].mode()
-      
-      encoded_output = jnp.transpose(encoded_output, (0, 2, 3, 4, 1))
 
       # Normalize latents
       latents_mean = jnp.array(self.vae.latents_mean).reshape(1, 1, 1, 1, self.vae.z_dim)