From d3cda804e709e914e074849325f2af96590b9ecf Mon Sep 17 00:00:00 2001 From: ginjia Date: Mon, 27 Nov 2023 20:32:43 +0800 Subject: [PATCH] add LoRA weights load and fuse support for IPEX pipeline (#5920) add IPEX pipeline LoRA weights loading support --- examples/community/stable_diffusion_ipex.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/community/stable_diffusion_ipex.py b/examples/community/stable_diffusion_ipex.py index 385227db0b..6d86248acb 100644 --- a/examples/community/stable_diffusion_ipex.py +++ b/examples/community/stable_diffusion_ipex.py @@ -21,7 +21,7 @@ from packaging import version from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer from diffusers.configuration_utils import FrozenDict -from diffusers.loaders import TextualInversionLoaderMixin +from diffusers.loaders import LoraLoaderMixin, TextualInversionLoaderMixin from diffusers.models import AutoencoderKL, UNet2DConditionModel from diffusers.pipelines.pipeline_utils import DiffusionPipeline from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput @@ -62,7 +62,7 @@ EXAMPLE_DOC_STRING = """ """ -class StableDiffusionIPEXPipeline(DiffusionPipeline, TextualInversionLoaderMixin): +class StableDiffusionIPEXPipeline(DiffusionPipeline, TextualInversionLoaderMixin, LoraLoaderMixin): r""" Pipeline for text-to-image generation using Stable Diffusion on IPEX.