removed redundant args

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p1.py

@@ -242,9 +242,7 @@ def _process_image_input(img_input, height, width, num_videos_per_prompt):
           prompt_embeds=prompt_embeds,
           negative_prompt_embeds=negative_prompt_embeds,
           image_embeds=image_embeds,
-          first_frame_mask=first_frame_mask,
           scheduler_state=scheduler_state,
-          rng=inference_rng,
       )
       latents = jnp.transpose(latents, (0, 4, 1, 2, 3))
       latents = self._denormalize_latents(latents)

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p2.py

@@ -236,7 +236,7 @@ def _process_image_input(img_input, height, width, num_videos_per_prompt):
           high_noise_graphdef=high_noise_graphdef, high_noise_state=high_noise_state, high_noise_rest=high_noise_rest,
           latents=latents, condition=condition,
           prompt_embeds=prompt_embeds, negative_prompt_embeds=negative_prompt_embeds,
-          scheduler_state=scheduler_state, rng=inference_rng,
+          scheduler_state=scheduler_state,
       )
       latents = jnp.transpose(latents, (0, 4, 1, 2, 3))
       latents = self._denormalize_latents(latents)