Skip to content

Commit 78c6e68

Browse files
authored
Merge branch 'main' into check-model-passed-pipeline
2 parents 2fd4508 + d041dd5 commit 78c6e68

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2313,7 +2313,7 @@ def _maybe_expand_transformer_param_shape_or_error_(
23132313
for name, module in transformer.named_modules():
23142314
if isinstance(module, torch.nn.Linear):
23152315
module_weight = module.weight.data
2316-
module_bias = module.bias.data if hasattr(module, "bias") else None
2316+
module_bias = module.bias.data if module.bias is not None else None
23172317
bias = module_bias is not None
23182318

23192319
lora_A_weight_name = f"{name}.lora_A.weight"

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)