Skip to content

Commit 659a272

Browse files
authored
Apply suggestions from code review
1 parent 0ba5e3e commit 659a272

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/pipelines/wan/pipeline_wan_i2v.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -695,7 +695,7 @@ def __call__(
695695
device, dtype=torch.float32
696696
)
697697

698-
prepared_latents = self.prepare_latents(
698+
latents_outputs = self.prepare_latents(
699699
image,
700700
batch_size * num_videos_per_prompt,
701701
num_channels_latents,
@@ -709,9 +709,9 @@ def __call__(
709709
last_image,
710710
)
711711
if self.config.expand_timesteps:
712-
latents, condition, first_frame_mask = prepared_latents
712+
latents, condition, first_frame_mask = latents_outputs
713713
else:
714-
latents, condition = prepared_latents
714+
latents, condition = latents_outputs
715715

716716
# 6. Denoising loop
717717
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order

0 commit comments

Comments
 (0)