From 33f936154db0bc7080960316b4ddb291e9555bf7 Mon Sep 17 00:00:00 2001 From: Teriks Date: Wed, 29 Jan 2025 11:18:47 -0600 Subject: [PATCH] support StableDiffusionAdapterPipeline.from_single_file (#10552) * support StableDiffusionAdapterPipeline.from_single_file * make style --------- Co-authored-by: Teriks Co-authored-by: hlky --- .../t2i_adapter/pipeline_stable_diffusion_adapter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py index 8520a2e2b7..6cd0e415e1 100644 --- a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +++ b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py @@ -22,7 +22,7 @@ import torch from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer from ...image_processor import VaeImageProcessor -from ...loaders import StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin +from ...loaders import FromSingleFileMixin, StableDiffusionLoraLoaderMixin, TextualInversionLoaderMixin from ...models import AutoencoderKL, MultiAdapter, T2IAdapter, UNet2DConditionModel from ...models.lora import adjust_lora_scale_text_encoder from ...schedulers import KarrasDiffusionSchedulers @@ -188,7 +188,7 @@ def retrieve_timesteps( return timesteps, num_inference_steps -class StableDiffusionAdapterPipeline(DiffusionPipeline, StableDiffusionMixin): +class StableDiffusionAdapterPipeline(DiffusionPipeline, StableDiffusionMixin, FromSingleFileMixin): r""" Pipeline for text-to-image generation using Stable Diffusion augmented with T2I-Adapter https://arxiv.org/abs/2302.08453