Skip to content

Commit 480510a

Browse files
authored
Change KolorsPipeline LoRA Loader to StableDiffusion (#11198)
Change LoRA Loader to StableDiffusion Replace the SDXL LoRA Loader Mixin inheritance with the StableDiffusion one
1 parent d9023a6 commit 480510a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/diffusers/pipelines/kolors/pipeline_kolors.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from ...callbacks import MultiPipelineCallbacks, PipelineCallback
2121
from ...image_processor import PipelineImageInput, VaeImageProcessor
22-
from ...loaders import IPAdapterMixin, StableDiffusionXLLoraLoaderMixin
22+
from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin
2323
from ...models import AutoencoderKL, ImageProjection, UNet2DConditionModel
2424
from ...models.attention_processor import AttnProcessor2_0, FusedAttnProcessor2_0, XFormersAttnProcessor
2525
from ...schedulers import KarrasDiffusionSchedulers
@@ -121,16 +121,16 @@ def retrieve_timesteps(
121121
return timesteps, num_inference_steps
122122

123123

124-
class KolorsPipeline(DiffusionPipeline, StableDiffusionMixin, StableDiffusionXLLoraLoaderMixin, IPAdapterMixin):
124+
class KolorsPipeline(DiffusionPipeline, StableDiffusionMixin, StableDiffusionLoraLoaderMixin, IPAdapterMixin):
125125
r"""
126126
Pipeline for text-to-image generation using Kolors.
127127
128128
This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods the
129129
library implements for all the pipelines (such as downloading or saving, running on a particular device, etc.)
130130
131131
The pipeline also inherits the following loading methods:
132-
- [`~loaders.StableDiffusionXLLoraLoaderMixin.load_lora_weights`] for loading LoRA weights
133-
- [`~loaders.StableDiffusionXLLoraLoaderMixin.save_lora_weights`] for saving LoRA weights
132+
- [`~loaders.StableDiffusionLoraLoaderMixin.load_lora_weights`] for loading LoRA weights
133+
- [`~loaders.StableDiffusionLoraLoaderMixin.save_lora_weights`] for saving LoRA weights
134134
- [`~loaders.IPAdapterMixin.load_ip_adapter`] for loading IP Adapters
135135
136136
Args:

0 commit comments

Comments
 (0)