Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py
View file @ 8ecf499d
...@@ -25,7 +25,7 @@ from transformers import ( ...@@ -25,7 +25,7 @@ from transformers import (
from ...callbacks import MultiPipelineCallbacks, PipelineCallback from ...callbacks import MultiPipelineCallbacks, PipelineCallback
from ...image_processor import PipelineImageInput, VaeImageProcessor from ...image_processor import PipelineImageInput, VaeImageProcessor
from ...loaders import SD3LoraLoaderMixin from ...loaders import FromSingleFileMixin, SD3LoraLoaderMixin
from ...models.autoencoders import AutoencoderKL from ...models.autoencoders import AutoencoderKL
from ...models.transformers import SD3Transformer2DModel from ...models.transformers import SD3Transformer2DModel
from ...schedulers import FlowMatchEulerDiscreteScheduler from ...schedulers import FlowMatchEulerDiscreteScheduler
...@@ -148,7 +148,7 @@ def retrieve_timesteps( ...@@ -148,7 +148,7 @@ def retrieve_timesteps(
return timesteps, num_inference_steps return timesteps, num_inference_steps
class StableDiffusion3InpaintPipeline(DiffusionPipeline): class StableDiffusion3InpaintPipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingleFileMixin):
r""" r"""
Args: Args:
transformer ([`SD3Transformer2DModel`]): 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