Skip to content

Commit d8d208a

Browse files
authored
Supper IP Adapter weight loading in StableDiffusionXLControlNetInpaintPipeline (#7031)
* support ip adapter loading * fix style
1 parent e0f33df commit d8d208a

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,12 @@
2828
)
2929

3030
from ...image_processor import PipelineImageInput, VaeImageProcessor
31-
from ...loaders import FromSingleFileMixin, StableDiffusionXLLoraLoaderMixin, TextualInversionLoaderMixin
31+
from ...loaders import (
32+
FromSingleFileMixin,
33+
IPAdapterMixin,
34+
StableDiffusionXLLoraLoaderMixin,
35+
TextualInversionLoaderMixin,
36+
)
3237
from ...models import AutoencoderKL, ControlNetModel, ImageProjection, UNet2DConditionModel
3338
from ...models.attention_processor import (
3439
AttnProcessor2_0,
@@ -146,7 +151,7 @@ def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0):
146151

147152

148153
class StableDiffusionXLControlNetInpaintPipeline(
149-
DiffusionPipeline, StableDiffusionXLLoraLoaderMixin, FromSingleFileMixin
154+
DiffusionPipeline, StableDiffusionXLLoraLoaderMixin, FromSingleFileMixin, IPAdapterMixin
150155
):
151156
r"""
152157
Pipeline for text-to-image generation using Stable Diffusion XL.
@@ -158,6 +163,7 @@ class StableDiffusionXLControlNetInpaintPipeline(
158163
- [`~loaders.StableDiffusionXLLoraLoaderMixin.load_lora_weights`] for loading LoRA weights
159164
- [`~loaders.StableDiffusionXLLoraLoaderMixin.save_lora_weights`] for saving LoRA weights
160165
- [`~loaders.FromSingleFileMixin.from_single_file`] for loading `.ckpt` files
166+
- [`~loaders.IPAdapterMixin.load_ip_adapter`] for loading IP Adapters
161167
162168
Args:
163169
vae ([`AutoencoderKL`]):

0 commit comments

Comments
 (0)