added changes for i2v 2.1

prishajain1 · prishajain1 · commit 4099d0b85197 · 2025-12-17T23:47:14.000+05:30
diff --git a/src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p1.py b/src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p1.py
@@ -163,8 +163,7 @@ def __call__(
 
     prompt_embeds, negative_prompt_embeds, image_embeds, effective_batch_size = self._prepare_model_inputs_i2v(
         prompt, image, negative_prompt, num_videos_per_prompt, max_sequence_length,
-        prompt_embeds, negative_prompt_embeds, image_embeds, last_image,
-        num_videos_per_prompt=num_videos_per_prompt,
+        prompt_embeds, negative_prompt_embeds, image_embeds, last_image
     )
 
     image_tensor = self.video_processor.preprocess(image, height=height, width=width)

Original file line number	Diff line number	Diff line change
`@@ -163,8 +163,7 @@ def __call__(`
`163`	`163`
`164`	`164`	`prompt_embeds, negative_prompt_embeds, image_embeds, effective_batch_size = self._prepare_model_inputs_i2v(`
`165`	`165`	`prompt, image, negative_prompt, num_videos_per_prompt, max_sequence_length,`
`166`		`- prompt_embeds, negative_prompt_embeds, image_embeds, last_image,`
`167`		`- num_videos_per_prompt=num_videos_per_prompt,`
	`166`	`+ prompt_embeds, negative_prompt_embeds, image_embeds, last_image`
`168`	`167`	`)`
`169`	`168`
`170`	`169`	`image_tensor = self.video_processor.preprocess(image, height=height, width=width)`