From 75124fc91ebb6007b364e37ab3589e66d3566355 Mon Sep 17 00:00:00 2001
From: Arpan Tripathi <42506819+tripathiarpan20@users.noreply.github.com>
Date: Thu, 15 Jun 2023 14:09:44 +0100
Subject: [PATCH] Added LoRA loading to `StableDiffusionKDiffusionPipeline`
 (#3751)

Added `LoraLoaderMixin` to `StableDiffusionKDiffusionPipeline`
---
 .../stable_diffusion/pipeline_stable_diffusion_k_diffusion.py   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
index ab613dd4df..e2800342e5 100755
--- 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.