Skip to content

Commit 18f9b99

Browse files
authored
Remove duplicate checks for len(generator) != batch_size when generator is a list (#10134)
remove duplicate checks
1 parent bf64b32 commit 18f9b99

File tree

3 files changed

+0
-18
lines changed

3 files changed

+0
-18
lines changed

src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -662,12 +662,6 @@ def prepare_latents(
662662
self.vae.to(dtype=torch.float32)
663663

664664
if isinstance(generator, list):
665-
if len(generator) != batch_size:
666-
raise ValueError(
667-
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
668-
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
669-
)
670-
671665
init_latents = [
672666
self.encode_video(video[i], generator[i], decode_chunk_size).unsqueeze(0)
673667
for i in range(batch_size)

src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -794,12 +794,6 @@ def prepare_latents(
794794
self.vae.to(dtype=torch.float32)
795795

796796
if isinstance(generator, list):
797-
if len(generator) != batch_size:
798-
raise ValueError(
799-
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
800-
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
801-
)
802-
803797
init_latents = [
804798
self.encode_video(video[i], generator[i], decode_chunk_size).unsqueeze(0)
805799
for i in range(batch_size)

src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -373,12 +373,6 @@ def prepare_latents(
373373

374374
if latents is None:
375375
if isinstance(generator, list):
376-
if len(generator) != batch_size:
377-
raise ValueError(
378-
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
379-
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
380-
)
381-
382376
init_latents = [
383377
retrieve_latents(self.vae.encode(video[i].unsqueeze(0)), generator[i]) for i in range(batch_size)
384378
]

0 commit comments

Comments
 (0)