Unverified Commit 8ecf499d authored by Eduardo Escobar's avatar Eduardo Escobar Committed by GitHub
Browse files

Enable `load_lora_weights` for `StableDiffusion3InpaintPipeline` (#9330)



Enable load_lora_weights for StableDiffusion3InpaintPipeline
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent dcf320f2
......@@ -25,7 +25,7 @@ from transformers import (
from ...callbacks import MultiPipelineCallbacks, PipelineCallback
from ...image_processor import PipelineImageInput, VaeImageProcessor
from ...loaders import SD3LoraLoaderMixin
from ...loaders import FromSingleFileMixin, SD3LoraLoaderMixin
from ...models.autoencoders import AutoencoderKL
from ...models.transformers import SD3Transformer2DModel
from ...schedulers import FlowMatchEulerDiscreteScheduler
......@@ -148,7 +148,7 @@ def retrieve_timesteps(
return timesteps, num_inference_steps
class StableDiffusion3InpaintPipeline(DiffusionPipeline):
class StableDiffusion3InpaintPipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingleFileMixin):
r"""
Args:
transformer ([`SD3Transformer2DModel`]):
......
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