AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Remove duplicate checks for len(generator) != batch_size when generator is a list (#10134)

Browse Source 
remove duplicate checks
This commit is contained in:
Aryan
2024-12-06 16:59:10 +05:30
committed by GitHub
parent bf64b32652
commit 18f9b99088
3 changed files with 0 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py
View File

@@ -662,12 +662,6 @@ class AnimateDiffVideoToVideoPipeline(
self.vae.to(dtype=torch.float32)
if isinstance(generator, list):
if len(generator) != batch_size:
raise ValueError(
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
)
init_latents = [
self.encode_video(video[i], generator[i], decode_chunk_size).unsqueeze(0)
for i in range(batch_size)

6
src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py
View File

@@ -794,12 +794,6 @@ class AnimateDiffVideoToVideoControlNetPipeline(
self.vae.to(dtype=torch.float32)
if isinstance(generator, list):
if len(generator) != batch_size:
raise ValueError(
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
)
init_latents = [
self.encode_video(video[i], generator[i], decode_chunk_size).unsqueeze(0)
for i in range(batch_size)

6
src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py
View File

@@ -373,12 +373,6 @@ class CogVideoXVideoToVideoPipeline(DiffusionPipeline, CogVideoXLoraLoaderMixin)
if latents is None:
if isinstance(generator, list):
if len(generator) != batch_size:
raise ValueError(
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
)
init_latents = [
retrieve_latents(self.vae.encode(video[i].unsqueeze(0)), generator[i]) for i in range(batch_size)
]