|
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