Skip to content

Commit d1411b3

Browse files
SahilCarterrsayakpaul
authored andcommitted
Added Error when len(gligen_images ) is not equal to len(gligen_phrases) in StableDiffusionGLIGENTextImagePipeline (#10176)
* added check value error * fix style
1 parent d38bca7 commit d1411b3

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

src/diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -446,13 +446,14 @@ def prepare_extra_step_kwargs(self, generator, eta):
446446
extra_step_kwargs["generator"] = generator
447447
return extra_step_kwargs
448448

449-
# Copied from diffusers.pipelines.stable_diffusion_k_diffusion.pipeline_stable_diffusion_k_diffusion.StableDiffusionKDiffusionPipeline.check_inputs
450449
def check_inputs(
451450
self,
452451
prompt,
453452
height,
454453
width,
455454
callback_steps,
455+
gligen_images,
456+
gligen_phrases,
456457
negative_prompt=None,
457458
prompt_embeds=None,
458459
negative_prompt_embeds=None,
@@ -499,6 +500,13 @@ def check_inputs(
499500
f" {negative_prompt_embeds.shape}."
500501
)
501502

503+
if gligen_images is not None and gligen_phrases is not None:
504+
if len(gligen_images) != len(gligen_phrases):
505+
raise ValueError(
506+
"`gligen_images` and `gligen_phrases` must have the same length when both are provided, but"
507+
f" got: `gligen_images` with length {len(gligen_images)} != `gligen_phrases` with length {len(gligen_phrases)}."
508+
)
509+
502510
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_latents
503511
def prepare_latents(self, batch_size, num_channels_latents, height, width, dtype, device, generator, latents=None):
504512
shape = (
@@ -814,6 +822,8 @@ def __call__(
814822
height,
815823
width,
816824
callback_steps,
825+
gligen_images,
826+
gligen_phrases,
817827
negative_prompt,
818828
prompt_embeds,
819829
negative_prompt_embeds,

0 commit comments

Comments
 (0)