Update src/diffusers/pipelines/wan/pipeline_wan_i2v.py

Co-authored-by: Aryan <[email protected]>

Author: yiyixuxu

src/diffusers/pipelines/wan/pipeline_wan_i2v.py

@@ -695,34 +695,23 @@ def __call__(
                 device, dtype=torch.float32
             )
 
+        prepared_latents = self.prepare_latents(
+            image,
+            batch_size * num_videos_per_prompt,
+            num_channels_latents,
+            height,
+            width,
+            num_frames,
+            torch.float32,
+            device,
+            generator,
+            latents,
+            last_image,
+        )
         if self.config.expand_timesteps:
-            latents, condition, first_frame_mask = self.prepare_latents(
-                image,
-                batch_size * num_videos_per_prompt,
-                num_channels_latents,
-                height,
-                width,
-                num_frames,
-                torch.float32,
-                device,
-                generator,
-                latents,
-                last_image,
-            )
+            latents, condition, first_frame_mask = prepared_latents
         else:
-            latents, condition = self.prepare_latents(
-                image,
-                batch_size * num_videos_per_prompt,
-                num_channels_latents,
-                height,
-                width,
-                num_frames,
-                torch.float32,
-                device,
-                generator,
-                latents,
-                last_image,
-            )
+            latents, condition = prepared_latents
 
         # 6. Denoising loop
         num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order