Skip to content

Commit d5bffb2

Browse files
committed
added check value error
1 parent c9e4fab commit d5bffb2

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

src/diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -446,13 +446,15 @@ 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
449+
450450
def check_inputs(
451451
self,
452452
prompt,
453453
height,
454454
width,
455455
callback_steps,
456+
gligen_images,
457+
gligen_phrases,
456458
negative_prompt=None,
457459
prompt_embeds=None,
458460
negative_prompt_embeds=None,
@@ -498,6 +500,15 @@ def check_inputs(
498500
f" got: `prompt_embeds` {prompt_embeds.shape} != `negative_prompt_embeds`"
499501
f" {negative_prompt_embeds.shape}."
500502
)
503+
504+
if gligen_images is not None and gligen_phrases is not None:
505+
if len(gligen_images) != len(gligen_phrases):
506+
raise ValueError(
507+
"`gligen_images` and `gligen_phrases` must have the same length when both are provided, but"
508+
f" got: `gligen_images` with length {len(gligen_images)} != `gligen_phrases` with length {len(gligen_phrases)}."
509+
)
510+
511+
501512

502513
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_latents
503514
def prepare_latents(self, batch_size, num_channels_latents, height, width, dtype, device, generator, latents=None):
@@ -814,6 +825,8 @@ def __call__(
814825
height,
815826
width,
816827
callback_steps,
828+
gligen_images,
829+
gligen_phrases,
817830
negative_prompt,
818831
prompt_embeds,
819832
negative_prompt_embeds,

0 commit comments

Comments
 (0)