randn_tensor argument passing changed

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p1.py

@@ -94,7 +94,7 @@ def prepare_latents(
         shape = (batch_size, num_channels_latents, num_latent_frames, latent_height, latent_width)
 
         if latents is None:
-            latents = randn_tensor(rng, shape, self.config, dtype)
+            latents = randn_tensor(shape, rng, self.config, dtype)
         else:
             latents = latents.astype(dtype)
         latent_condition, _ = self.prepare_latents_i2v_base(image, num_frames, dtype, last_image)

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p2.py

@@ -93,7 +93,7 @@ def prepare_latents(
     shape = (batch_size, num_channels_latents, num_latent_frames, latent_height, latent_width)
 
     if latents is None:
-        latents = randn_tensor(rng, shape, self.config, dtype)
+        latents = randn_tensor(shape, rng, self.config, dtype)
     else:
         latents = latents.astype(dtype)