Skip to content

Commit 8ecf499

Browse files
baldescosayakpaul
andauthored
Enable load_lora_weights for StableDiffusion3InpaintPipeline (#9330)
Enable load_lora_weights for StableDiffusion3InpaintPipeline Co-authored-by: Sayak Paul <[email protected]>
1 parent dcf320f commit 8ecf499

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525

2626
from ...callbacks import MultiPipelineCallbacks, PipelineCallback
2727
from ...image_processor import PipelineImageInput, VaeImageProcessor
28-
from ...loaders import SD3LoraLoaderMixin
28+
from ...loaders import FromSingleFileMixin, SD3LoraLoaderMixin
2929
from ...models.autoencoders import AutoencoderKL
3030
from ...models.transformers import SD3Transformer2DModel
3131
from ...schedulers import FlowMatchEulerDiscreteScheduler
@@ -148,7 +148,7 @@ def retrieve_timesteps(
148148
return timesteps, num_inference_steps
149149

150150

151-
class StableDiffusion3InpaintPipeline(DiffusionPipeline):
151+
class StableDiffusion3InpaintPipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingleFileMixin):
152152
r"""
153153
Args:
154154
transformer ([`SD3Transformer2DModel`]):

0 commit comments

Comments
 (0)