Skip to content

Commit 33c7cd6

Browse files
committed
make style
1 parent 4f51829 commit 33c7cd6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -363,7 +363,7 @@ def prepare_latents(
363363
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
364364
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
365365
)
366-
366+
367367
assert image.ndim == 4
368368
image = image.unsqueeze(2) # [B, C, F, H, W]
369369
print(image.shape)
@@ -755,7 +755,7 @@ def __call__(
755755

756756
latent_model_input = torch.cat([latents] * 2) if do_classifier_free_guidance else latents
757757
latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
758-
758+
759759
latent_image_input = torch.cat([image_latents] * 2) if do_classifier_free_guidance else image_latents
760760
latent_model_input = torch.cat([latent_model_input, latent_image_input], dim=2)
761761
print(latent_model_input.shape)

0 commit comments

Comments
 (0)