Skip to content

Commit 8af793b

Browse files
Adding TextualInversionLoaderMixin for the controlnet_inpaint_sd_xl pipeline (#7288)
* added TextualInversionMixIn to controlnet_inpaint_sd_xl pipeline --------- Co-authored-by: YiYi Xu <[email protected]>
1 parent eb96ff0 commit 8af793b

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,12 @@ def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0):
151151

152152

153153
class StableDiffusionXLControlNetInpaintPipeline(
154-
DiffusionPipeline, StableDiffusionMixin, StableDiffusionXLLoraLoaderMixin, FromSingleFileMixin, IPAdapterMixin
154+
DiffusionPipeline,
155+
StableDiffusionMixin,
156+
StableDiffusionXLLoraLoaderMixin,
157+
FromSingleFileMixin,
158+
IPAdapterMixin,
159+
TextualInversionLoaderMixin,
155160
):
156161
r"""
157162
Pipeline for text-to-image generation using Stable Diffusion XL.
@@ -160,6 +165,7 @@ class StableDiffusionXLControlNetInpaintPipeline(
160165
library implements for all the pipelines (such as downloading or saving, running on a particular device, etc.)
161166
162167
The pipeline also inherits the following loading methods:
168+
- [`~loaders.TextualInversionLoaderMixin.load_textual_inversion`] for loading textual inversion embeddings
163169
- [`~loaders.StableDiffusionXLLoraLoaderMixin.load_lora_weights`] for loading LoRA weights
164170
- [`~loaders.StableDiffusionXLLoraLoaderMixin.save_lora_weights`] for saving LoRA weights
165171
- [`~loaders.FromSingleFileMixin.from_single_file`] for loading `.ckpt` files

0 commit comments

Comments
 (0)