Skip to content

GitLab

Unverified Commit d3cda804 authored by ginjia's avatar ginjia Committed by GitHub
Browse files

add LoRA weights load and fuse support for IPEX pipeline (#5920) 

add IPEX pipeline LoRA weights loading support
parent 07eac4d6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/community/stable_diffusion_ipex.py
View file @ d3cda804
...@@ -21,7 +21,7 @@ from packaging import version ...@@ -21,7 +21,7 @@ from packaging import version
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
from diffusers.configuration_utils import FrozenDict 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.models import AutoencoderKL, UNet2DConditionModel
from diffusers.pipelines.pipeline_utils import DiffusionPipeline from diffusers.pipelines.pipeline_utils import DiffusionPipeline
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
...@@ -62,7 +62,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -62,7 +62,7 @@ EXAMPLE_DOC_STRING = """
""" """
class StableDiffusionIPEXPipeline(DiffusionPipeline, TextualInversionLoaderMixin): class StableDiffusionIPEXPipeline(DiffusionPipeline, TextualInversionLoaderMixin, LoraLoaderMixin):
r""" r"""
Pipeline for text-to-image generation using Stable Diffusion on IPEX. Pipeline for text-to-image generation using Stable Diffusion on IPEX.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment