Added LoRA loading to `StableDiffusionKDiffusionPipeline` (#3751)

Added `LoraLoaderMixin` to `StableDiffusionKDiffusionPipeline`

Added LoRA loading to `StableDiffusionKDiffusionPipeline` (#3751)
Added `LoraLoaderMixin` to `StableDiffusionKDiffusionPipeline`
75124fc9 · Arpan Tripathi · GitHub · 908e5e9c · 75124fc9
Unverified Commit 75124fc9 authored Jun 15, 2023 by Arpan Tripathi Committed by GitHub Jun 15, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py ...stable_diffusion/pipeline_stable_diffusion_k_diffusion.py +1 -1

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
+++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
@@ -45,7 +45,7 @@ class ModelWrapper:
        return self.model(*args, encoder_hidden_states=encoder_hidden_states, **kwargs).sample


-class StableDiffusionKDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderMixin):
+class StableDiffusionKDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderMixin, LoraLoaderMixin):
    r"""
    Pipeline for text-to-image generation using Stable Diffusion.