|
19 | 19 |
|
20 | 20 | from ...callbacks import MultiPipelineCallbacks, PipelineCallback
|
21 | 21 | from ...image_processor import PipelineImageInput, VaeImageProcessor
|
22 |
| -from ...loaders import IPAdapterMixin, StableDiffusionXLLoraLoaderMixin |
| 22 | +from ...loaders import IPAdapterMixin, StableDiffusionLoraLoaderMixin |
23 | 23 | from ...models import AutoencoderKL, ImageProjection, UNet2DConditionModel
|
24 | 24 | from ...models.attention_processor import AttnProcessor2_0, FusedAttnProcessor2_0, XFormersAttnProcessor
|
25 | 25 | from ...schedulers import KarrasDiffusionSchedulers
|
@@ -121,16 +121,16 @@ def retrieve_timesteps(
|
121 | 121 | return timesteps, num_inference_steps
|
122 | 122 |
|
123 | 123 |
|
124 |
| -class KolorsPipeline(DiffusionPipeline, StableDiffusionMixin, StableDiffusionXLLoraLoaderMixin, IPAdapterMixin): |
| 124 | +class KolorsPipeline(DiffusionPipeline, StableDiffusionMixin, StableDiffusionLoraLoaderMixin, IPAdapterMixin): |
125 | 125 | r"""
|
126 | 126 | Pipeline for text-to-image generation using Kolors.
|
127 | 127 |
|
128 | 128 | This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods the
|
129 | 129 | library implements for all the pipelines (such as downloading or saving, running on a particular device, etc.)
|
130 | 130 |
|
131 | 131 | 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 |
134 | 134 | - [`~loaders.IPAdapterMixin.load_ip_adapter`] for loading IP Adapters
|
135 | 135 |
|
136 | 136 | Args:
|
|
0 commit comments