From 16f28a941083d915e0f44aa1e74af32c1bdf6344 Mon Sep 17 00:00:00 2001 From: ishan-modi Date: Tue, 29 Apr 2025 16:21:48 +0530 Subject: [PATCH 1/2] update --- src/diffusers/pipelines/allegro/pipeline_allegro.py | 6 ++++-- .../pipelines/animatediff/pipeline_animatediff.py | 5 ++++- .../animatediff/pipeline_animatediff_controlnet.py | 5 ++++- .../pipelines/animatediff/pipeline_animatediff_sdxl.py | 7 ++++++- .../animatediff/pipeline_animatediff_sparsectrl.py | 3 +++ .../animatediff/pipeline_animatediff_video2video.py | 5 ++++- .../pipeline_animatediff_video2video_controlnet.py | 5 ++++- src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py | 3 +++ src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py | 7 +++++-- .../pipelines/cogvideo/pipeline_cogvideox_fun_control.py | 7 +++++-- .../pipelines/cogvideo/pipeline_cogvideox_image2video.py | 7 +++++-- .../pipelines/cogvideo/pipeline_cogvideox_video2video.py | 7 +++++-- src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py | 4 +++- src/diffusers/pipelines/cogview4/pipeline_cogview4.py | 7 +++++-- .../pipelines/cogview4/pipeline_cogview4_control.py | 7 +++++-- src/diffusers/pipelines/consisid/pipeline_consisid.py | 5 ++++- src/diffusers/pipelines/controlnet/pipeline_controlnet.py | 5 ++++- .../pipelines/controlnet/pipeline_controlnet_img2img.py | 5 ++++- .../pipelines/controlnet/pipeline_controlnet_inpaint.py | 5 ++++- .../controlnet/pipeline_controlnet_inpaint_sd_xl.py | 5 ++++- .../pipelines/controlnet/pipeline_controlnet_sd_xl.py | 6 +++++- .../controlnet/pipeline_controlnet_sd_xl_img2img.py | 5 ++++- .../controlnet/pipeline_controlnet_union_inpaint_sd_xl.py | 5 ++++- .../controlnet/pipeline_controlnet_union_sd_xl.py | 6 +++++- .../controlnet/pipeline_controlnet_union_sd_xl_img2img.py | 5 ++++- .../pipeline_hunyuandit_controlnet.py | 5 ++++- .../pipeline_stable_diffusion_3_controlnet.py | 5 ++++- .../pipeline_stable_diffusion_3_controlnet_inpainting.py | 5 ++++- .../pipelines/controlnet_xs/pipeline_controlnet_xs.py | 4 +++- .../controlnet_xs/pipeline_controlnet_xs_sd_xl.py | 4 +++- .../deprecated/alt_diffusion/pipeline_alt_diffusion.py | 4 ++++ .../alt_diffusion/pipeline_alt_diffusion_img2img.py | 3 +++ .../pipelines/easyanimate/pipeline_easyanimate.py | 5 ++++- .../pipelines/easyanimate/pipeline_easyanimate_control.py | 5 ++++- .../pipelines/easyanimate/pipeline_easyanimate_inpaint.py | 5 ++++- src/diffusers/pipelines/flux/pipeline_flux.py | 7 +++++-- src/diffusers/pipelines/flux/pipeline_flux_control.py | 5 ++++- .../pipelines/flux/pipeline_flux_control_img2img.py | 5 ++++- .../pipelines/flux/pipeline_flux_control_inpaint.py | 5 ++++- src/diffusers/pipelines/flux/pipeline_flux_controlnet.py | 5 ++++- .../flux/pipeline_flux_controlnet_image_to_image.py | 5 ++++- .../pipelines/flux/pipeline_flux_controlnet_inpainting.py | 5 ++++- src/diffusers/pipelines/flux/pipeline_flux_fill.py | 5 ++++- src/diffusers/pipelines/flux/pipeline_flux_img2img.py | 5 ++++- src/diffusers/pipelines/flux/pipeline_flux_inpaint.py | 5 ++++- .../pipelines/hidream_image/pipeline_hidream_image.py | 5 ++++- .../pipeline_hunyuan_skyreels_image2video.py | 7 +++++-- .../pipelines/hunyuan_video/pipeline_hunyuan_video.py | 7 +++++-- .../hunyuan_video/pipeline_hunyuan_video_image2video.py | 7 +++++-- src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py | 5 ++++- src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py | 1 + .../pipelines/kandinsky2_2/pipeline_kandinsky2_2.py | 2 ++ .../kandinsky2_2/pipeline_kandinsky2_2_img2img.py | 2 ++ .../kandinsky2_2/pipeline_kandinsky2_2_inpainting.py | 2 ++ .../pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py | 2 ++ src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py | 2 ++ .../pipelines/kandinsky3/pipeline_kandinsky3_img2img.py | 2 ++ src/diffusers/pipelines/kolors/pipeline_kolors.py | 6 +++++- src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py | 7 ++++++- .../pipeline_latent_consistency_img2img.py | 3 +++ .../pipeline_latent_consistency_text2img.py | 3 +++ src/diffusers/pipelines/latte/pipeline_latte.py | 6 ++++-- .../ledits_pp/pipeline_leditspp_stable_diffusion.py | 2 ++ .../ledits_pp/pipeline_leditspp_stable_diffusion_xl.py | 5 +++++ src/diffusers/pipelines/ltx/pipeline_ltx.py | 7 +++++-- src/diffusers/pipelines/ltx/pipeline_ltx_condition.py | 7 +++++-- src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py | 7 +++++-- src/diffusers/pipelines/lumina/pipeline_lumina.py | 2 ++ src/diffusers/pipelines/lumina2/pipeline_lumina2.py | 3 +++ src/diffusers/pipelines/mochi/pipeline_mochi.py | 7 +++++-- src/diffusers/pipelines/omnigen/pipeline_omnigen.py | 4 +++- src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py | 3 +++ .../pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py | 3 +++ .../pipelines/pag/pipeline_pag_controlnet_sd_xl.py | 4 ++++ .../pag/pipeline_pag_controlnet_sd_xl_img2img.py | 3 +++ src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py | 5 ++++- src/diffusers/pipelines/pag/pipeline_pag_kolors.py | 6 +++++- src/diffusers/pipelines/pag/pipeline_pag_sana.py | 4 +++- src/diffusers/pipelines/pag/pipeline_pag_sd.py | 6 +++++- src/diffusers/pipelines/pag/pipeline_pag_sd_3.py | 5 ++++- src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py | 5 ++++- .../pipelines/pag/pipeline_pag_sd_animatediff.py | 3 +++ src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py | 5 ++++- src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py | 6 +++++- src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py | 7 ++++++- src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py | 8 +++++++- src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py | 8 +++++++- src/diffusers/pipelines/pia/pipeline_pia.py | 3 +++ src/diffusers/pipelines/sana/pipeline_sana.py | 5 ++++- src/diffusers/pipelines/sana/pipeline_sana_controlnet.py | 5 ++++- src/diffusers/pipelines/sana/pipeline_sana_sprint.py | 5 ++++- .../pipelines/stable_cascade/pipeline_stable_cascade.py | 2 ++ .../stable_cascade/pipeline_stable_cascade_prior.py | 2 ++ .../stable_diffusion/pipeline_stable_diffusion.py | 6 +++++- .../pipeline_stable_diffusion_depth2img.py | 3 +++ .../stable_diffusion/pipeline_stable_diffusion_img2img.py | 5 ++++- .../stable_diffusion/pipeline_stable_diffusion_inpaint.py | 5 ++++- .../pipeline_stable_diffusion_instruct_pix2pix.py | 2 ++ .../stable_diffusion_3/pipeline_stable_diffusion_3.py | 5 ++++- .../pipeline_stable_diffusion_3_img2img.py | 5 ++++- .../pipeline_stable_diffusion_3_inpaint.py | 5 ++++- .../pipeline_stable_diffusion_xl_k_diffusion.py | 1 + .../pipeline_stable_diffusion_ldm3d.py | 6 +++++- .../pipeline_stable_diffusion_panorama.py | 6 +++++- .../stable_diffusion_xl/pipeline_stable_diffusion_xl.py | 7 ++++++- .../pipeline_stable_diffusion_xl_img2img.py | 8 +++++++- .../pipeline_stable_diffusion_xl_inpaint.py | 8 +++++++- .../pipeline_stable_video_diffusion.py | 2 ++ .../t2i_adapter/pipeline_stable_diffusion_adapter.py | 1 + .../t2i_adapter/pipeline_stable_diffusion_xl_adapter.py | 1 + src/diffusers/pipelines/wan/pipeline_wan.py | 7 +++++-- src/diffusers/pipelines/wan/pipeline_wan_i2v.py | 7 +++++-- src/diffusers/pipelines/wan/pipeline_wan_video2video.py | 7 +++++-- src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py | 2 ++ .../pipelines/wuerstchen/pipeline_wuerstchen_prior.py | 2 ++ 115 files changed, 450 insertions(+), 101 deletions(-) diff --git a/src/diffusers/pipelines/allegro/pipeline_allegro.py b/src/diffusers/pipelines/allegro/pipeline_allegro.py index 1fc1c0901ab3..5c368dc2b23a 100644 --- a/src/diffusers/pipelines/allegro/pipeline_allegro.py +++ b/src/diffusers/pipelines/allegro/pipeline_allegro.py @@ -198,6 +198,10 @@ def __init__( scheduler: KarrasDiffusionSchedulers, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -819,8 +823,6 @@ def __call__( negative_prompt_attention_mask, ) self._guidance_scale = guidance_scale - self._current_timestep = None - self._interrupt = False # 2. Default height and width to transformer if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff.py index d3ad5cc13ce3..da9c418212a2 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff.py @@ -136,6 +136,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) @@ -714,7 +718,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, (str, dict)): diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py index db546398643b..97aef02544a0 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py @@ -172,6 +172,10 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) @@ -877,7 +881,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, (str, dict)): diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py index 958eb5fb5134..7ede5b454c0d 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py @@ -299,6 +299,12 @@ def __init__( force_zeros_for_empty_prompt: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) @@ -1082,7 +1088,6 @@ def __call__( self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py index 8c51fddcd5fc..a1c4eb9605a3 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py @@ -184,6 +184,9 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py index 116397055272..17edd1e483f5 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py @@ -239,6 +239,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) @@ -883,7 +887,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, (str, dict)): diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py index ce974094936a..22d42662060f 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py @@ -262,6 +262,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) @@ -1086,7 +1090,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, (str, dict)): diff --git a/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py b/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py index 7c98b3b71c48..3edbfa81c804 100644 --- a/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py +++ b/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py @@ -154,6 +154,9 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._attention_kwargs=None + self._num_timesteps=0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py index 99ae9025cd3e..39b9864af3c3 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py @@ -185,6 +185,11 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -631,8 +636,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False # 2. Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py index e37574ec9cb2..533521d43783 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py @@ -192,6 +192,11 @@ def __init__( scheduler: KarrasDiffusionSchedulers, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -684,8 +689,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False # 2. Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py index 59d7c4cad547..0816e3f334b5 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py @@ -202,6 +202,11 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, @@ -732,9 +737,7 @@ def __call__( negative_prompt_embeds=negative_prompt_embeds, ) self._guidance_scale = guidance_scale - self._current_timestep = None self._attention_kwargs = attention_kwargs - self._interrupt = False # 2. Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py index c4dc7e574f7e..2024fe4dd8d4 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py @@ -207,6 +207,11 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -704,8 +709,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False # 2. Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py index 0cd3943fbcd2..8aa690a5a017 100644 --- a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py +++ b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py @@ -156,6 +156,9 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -534,7 +537,6 @@ def __call__( negative_prompt_embeds, ) self._guidance_scale = guidance_scale - self._interrupt = False # 2. Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py index 8550fa94f9e4..4205a1876b4d 100644 --- a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py +++ b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py @@ -168,6 +168,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -535,8 +540,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False # Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py b/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py index 7613bc3d0f40..a352c0e05f68 100644 --- a/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py +++ b/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py @@ -170,6 +170,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -567,8 +572,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False # Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/consisid/pipeline_consisid.py b/src/diffusers/pipelines/consisid/pipeline_consisid.py index 8abe24e80786..e09e11f38fa9 100644 --- a/src/diffusers/pipelines/consisid/pipeline_consisid.py +++ b/src/diffusers/pipelines/consisid/pipeline_consisid.py @@ -288,6 +288,10 @@ def __init__( scheduler: CogVideoXDPMScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, @@ -808,7 +812,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False # 2. Default call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet.py index a5e38278cdf2..e8e6c04c2552 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet.py @@ -223,6 +223,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if safety_checker is None and requires_safety_checker: logger.warning( @@ -1100,7 +1104,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py index be2874f48e69..eb7f5c6bc356 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py @@ -201,6 +201,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if safety_checker is None and requires_safety_checker: logger.warning( @@ -1092,7 +1096,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py index 2c63aedd966b..4a89dfb22709 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py @@ -207,6 +207,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if safety_checker is None and requires_safety_checker: logger.warning( @@ -1209,7 +1213,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py index 5907b41f4e73..b55163a746a2 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py @@ -257,6 +257,10 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) @@ -1457,7 +1461,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py index 77d496cf831d..691243ab6624 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py @@ -269,6 +269,11 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) @@ -1262,7 +1267,6 @@ def __call__( self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py index 04f069e12eb9..5a6d1c56221a 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py @@ -262,6 +262,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) @@ -1355,7 +1359,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py index 8aae9ee7a281..2e83c1b02209 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py @@ -239,6 +239,10 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if not isinstance(controlnet, ControlNetUnionModel): raise ValueError("Expected `controlnet` to be of type `ControlNetUnionModel`.") @@ -1395,7 +1399,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py index ca931c221eec..4bbe071ea396 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py @@ -259,6 +259,11 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetUnionModel(controlnet) @@ -1249,7 +1254,6 @@ def __call__( self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py index 87398395d99e..bdef279be33b 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py @@ -271,6 +271,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if not isinstance(controlnet, ControlNetUnionModel): raise ValueError("Expected `controlnet` to be of type `ControlNetUnionModel`.") @@ -1322,7 +1326,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py b/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py index 5ee712b5f116..494172f7119c 100644 --- a/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py +++ b/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py @@ -237,6 +237,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = HunyuanDiT2DMultiControlNetModel(controlnet) @@ -801,7 +805,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._guidance_rescale = guidance_rescale - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py index 7f7acd882b59..9c05a2793ac4 100644 --- a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +++ b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py @@ -206,6 +206,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = SD3MultiControlNetModel(controlnet) if isinstance(controlnet, SD3MultiControlNetModel): @@ -1010,7 +1014,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py index cb35f67fa112..e0c737ff5e5d 100644 --- a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py +++ b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py @@ -227,6 +227,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1059,7 +1063,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py index 901ca25c576c..8776f679a988 100644 --- a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +++ b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py @@ -156,6 +156,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if isinstance(unet, UNet2DConditionModel): unet = UNetControlNetXSModel.from_unet(unet, controlnet) @@ -771,7 +774,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py index acf1f5489ec1..e2bd56990792 100644 --- a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +++ b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py @@ -191,6 +191,9 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if isinstance(unet, UNet2DConditionModel): unet = UNetControlNetXSModel.from_unet(unet, controlnet) @@ -915,7 +918,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py index 48c0aa4f6d76..1e073d98e162 100644 --- a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +++ b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py @@ -209,6 +209,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py index fa70689d790d..8f474c2874bd 100644 --- a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +++ b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py @@ -237,6 +237,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py index 25975b04f395..f0d70ff445f1 100755 --- a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py +++ b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py @@ -217,6 +217,10 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -630,7 +634,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._guidance_rescale = guidance_rescale - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py index 1d2c508675f1..65dbb27cd9b1 100755 --- a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py +++ b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py @@ -319,6 +319,10 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -773,7 +777,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._guidance_rescale = guidance_rescale - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py index aaec454cc723..671e463095a4 100755 --- a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py +++ b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py @@ -361,6 +361,10 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -919,7 +923,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._guidance_rescale = guidance_rescale - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/flux/pipeline_flux.py b/src/diffusers/pipelines/flux/pipeline_flux.py index a7266c3a565e..d6b39beb42d0 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux.py +++ b/src/diffusers/pipelines/flux/pipeline_flux.py @@ -193,6 +193,11 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._current_timestep=None + self._interrupt=False self.register_modules( vae=vae, @@ -775,8 +780,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._current_timestep = None - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control.py b/src/diffusers/pipelines/flux/pipeline_flux_control.py index 113b0dd7291f..6d39491f9403 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_control.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_control.py @@ -203,6 +203,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -726,7 +730,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py b/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py index c269be15a4b2..cb3c45af4681 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py @@ -217,6 +217,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -765,7 +769,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Preprocess image init_image = self.image_processor.preprocess(image, height=height, width=width) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py b/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py index af7e8b53fad3..6d5ce5391605 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py @@ -248,6 +248,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -923,7 +927,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False device = self._execution_device # 3. Define call parameters diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py index f3f1d90204d6..4482c22c679c 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py @@ -221,6 +221,10 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = FluxMultiControlNetModel(controlnet) @@ -842,7 +846,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py index cd9508c00d99..250d6f3e5a86 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py @@ -214,6 +214,10 @@ def __init__( ], ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = FluxMultiControlNetModel(controlnet) @@ -753,7 +757,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False if prompt is not None and isinstance(prompt, str): batch_size = 1 diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py index 64cc8e13f33f..153671b1562a 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py @@ -216,6 +216,10 @@ def __init__( ], ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if isinstance(controlnet, (list, tuple)): controlnet = FluxMultiControlNetModel(controlnet) @@ -879,7 +883,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/flux/pipeline_flux_fill.py b/src/diffusers/pipelines/flux/pipeline_flux_fill.py index 546a225aa999..f0c8dbf6fb35 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_fill.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_fill.py @@ -211,6 +211,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -863,7 +867,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False init_image = self.image_processor.preprocess(image, height=height, width=width) init_image = init_image.to(dtype=torch.float32) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_img2img.py b/src/diffusers/pipelines/flux/pipeline_flux_img2img.py index 64cd6ac45f1a..78499a074af0 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_img2img.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_img2img.py @@ -210,6 +210,10 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -822,7 +826,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Preprocess image init_image = self.image_processor.preprocess(image, height=height, width=width) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py b/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py index ecd5a8967b4f..3a49cb5e3fbb 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py @@ -207,6 +207,10 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -945,7 +949,6 @@ def __call__( self._guidance_scale = guidance_scale self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Preprocess mask and image if padding_mask_crop is not None: diff --git a/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py b/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py index 6fe74cbd9acc..bc6faa491bea 100644 --- a/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py +++ b/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py @@ -162,6 +162,10 @@ def __init__( transformer: HiDreamImageTransformer2DModel, ): super().__init__() + self._guidance_scale=1.0 + self._attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -863,7 +867,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py index 297d2a9c9396..a45bacb003d0 100644 --- a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py +++ b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py @@ -206,6 +206,11 @@ def __init__( tokenizer_2: CLIPTokenizer, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( vae=vae, @@ -656,8 +661,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False device = self._execution_device diff --git a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py index 3cb91b3782f2..c14dc3d3b392 100644 --- a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py +++ b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py @@ -181,6 +181,11 @@ def __init__( tokenizer_2: CLIPTokenizer, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( vae=vae, @@ -612,8 +617,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False device = self._execution_device diff --git a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py index 18a0e970c610..39caccb255b9 100644 --- a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py +++ b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py @@ -259,6 +259,11 @@ def __init__( image_processor: CLIPImageProcessor, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( vae=vae, @@ -802,8 +807,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False device = self._execution_device diff --git a/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py b/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py index febf2b0392cc..4c98bd5de047 100644 --- a/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py +++ b/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py @@ -211,6 +211,10 @@ def __init__( tokenizer_2: Optional[MT5Tokenizer] = None, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -717,7 +721,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._guidance_rescale = guidance_rescale - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py b/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py index 58d65a190d5b..c941463ea26c 100644 --- a/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py +++ b/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py @@ -132,6 +132,7 @@ def __init__( scheduler: DDIMScheduler, ): super().__init__() + self._guidance_scale=1.0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py index a584674540d8..42421716c4b8 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py @@ -95,6 +95,8 @@ def __init__( movq: VQModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( unet=unet, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py index 909c6375672b..9bd3cf4534f8 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py @@ -101,6 +101,8 @@ def __init__( movq: VQModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( unet=unet, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py index 482093a4bb29..0d532d367604 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py @@ -266,6 +266,8 @@ def __init__( movq: VQModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( unet=unet, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py index d05a7fbdb1b8..3207762ff59d 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py @@ -127,6 +127,8 @@ def __init__( image_processor: CLIPImageProcessor, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( prior=prior, diff --git a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py index 5309f94a53c8..dc7347b64ab9 100644 --- a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +++ b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py @@ -75,6 +75,8 @@ def __init__( movq: VQModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, unet=unet, scheduler=scheduler, movq=movq diff --git a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py index f8fa84085c58..20af7f9e886a 100644 --- a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +++ b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py @@ -72,6 +72,8 @@ def __init__( movq: VQModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, unet=unet, scheduler=scheduler, movq=movq diff --git a/src/diffusers/pipelines/kolors/pipeline_kolors.py b/src/diffusers/pipelines/kolors/pipeline_kolors.py index 1fc4c02cc43f..581ae51ee419 100644 --- a/src/diffusers/pipelines/kolors/pipeline_kolors.py +++ b/src/diffusers/pipelines/kolors/pipeline_kolors.py @@ -177,6 +177,11 @@ def __init__( force_zeros_for_empty_prompt: bool = False, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -857,7 +862,6 @@ def __call__( self._guidance_scale = guidance_scale self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py b/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py index df94ec3f0f24..c8cbfd033338 100644 --- a/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py +++ b/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py @@ -196,6 +196,12 @@ def __init__( force_zeros_for_empty_prompt: bool = False, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._denoising_start=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1010,7 +1016,6 @@ def __call__( self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end self._denoising_start = denoising_start - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py index 1c59ca7d6d7c..d5ae98fcbb14 100644 --- a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +++ b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py @@ -212,6 +212,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py index a3d9917d3376..46195876298a 100644 --- a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +++ b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py @@ -191,6 +191,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/latte/pipeline_latte.py b/src/diffusers/pipelines/latte/pipeline_latte.py index 977e648d85a3..67d01c9663ca 100644 --- a/src/diffusers/pipelines/latte/pipeline_latte.py +++ b/src/diffusers/pipelines/latte/pipeline_latte.py @@ -185,6 +185,10 @@ def __init__( scheduler: KarrasDiffusionSchedulers, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -733,8 +737,6 @@ def __call__( negative_prompt_embeds, ) self._guidance_scale = guidance_scale - self._current_timestep = None - self._interrupt = False # 2. Default height and width to transformer if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py index bdac47c47ade..1d122f1bc878 100644 --- a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +++ b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py @@ -315,6 +315,8 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None if not isinstance(scheduler, DDIMScheduler) and not isinstance(scheduler, DPMSolverMultistepScheduler): scheduler = DPMSolverMultistepScheduler.from_config( diff --git a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py index cad7d8a66a08..bd987b82e944 100644 --- a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +++ b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py @@ -359,6 +359,11 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx.py b/src/diffusers/pipelines/ltx/pipeline_ltx.py index 6f3faed8ff72..6dd1d6e9a51e 100644 --- a/src/diffusers/pipelines/ltx/pipeline_ltx.py +++ b/src/diffusers/pipelines/ltx/pipeline_ltx.py @@ -177,6 +177,11 @@ def __init__( transformer: LTXVideoTransformer3DModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._attention_kwargs=None + self._interrupt=False self.register_modules( vae=vae, @@ -625,8 +630,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False - self._current_timestep = None # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py b/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py index dcfdfaf23288..a2a50085794b 100644 --- a/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py +++ b/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py @@ -259,6 +259,11 @@ def __init__( transformer: LTXVideoTransformer3DModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._attention_kwargs=None + self._interrupt=False self.register_modules( vae=vae, @@ -940,8 +945,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False - self._current_timestep = None # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py b/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py index 1ae67967c6f5..51b1b9a5c3f0 100644 --- a/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py +++ b/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py @@ -196,6 +196,11 @@ def __init__( transformer: LTXVideoTransformer3DModel, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._attention_kwargs=None + self._interrupt=False self.register_modules( vae=vae, @@ -689,8 +694,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False - self._current_timestep = None # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/lumina/pipeline_lumina.py b/src/diffusers/pipelines/lumina/pipeline_lumina.py index 22ff926afaff..cff89812db3f 100644 --- a/src/diffusers/pipelines/lumina/pipeline_lumina.py +++ b/src/diffusers/pipelines/lumina/pipeline_lumina.py @@ -185,6 +185,8 @@ def __init__( tokenizer: Union[GemmaTokenizer, GemmaTokenizerFast], ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/lumina2/pipeline_lumina2.py b/src/diffusers/pipelines/lumina2/pipeline_lumina2.py index e0905a2f131f..54ba1c6f64b9 100644 --- a/src/diffusers/pipelines/lumina2/pipeline_lumina2.py +++ b/src/diffusers/pipelines/lumina2/pipeline_lumina2.py @@ -167,6 +167,9 @@ def __init__( tokenizer: Union[GemmaTokenizer, GemmaTokenizerFast], ): super().__init__() + self._guidance_scale=1.0 + self._attention_kwargs=None + self._num_timesteps=0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/mochi/pipeline_mochi.py b/src/diffusers/pipelines/mochi/pipeline_mochi.py index d1f88b02c5cc..d368c3d0cb66 100644 --- a/src/diffusers/pipelines/mochi/pipeline_mochi.py +++ b/src/diffusers/pipelines/mochi/pipeline_mochi.py @@ -177,6 +177,11 @@ def __init__( force_zeros_for_empty_prompt: bool = False, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._attention_kwargs=None + self._current_timestep=None + self._interrupt=False self.register_modules( vae=vae, @@ -594,8 +599,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/omnigen/pipeline_omnigen.py b/src/diffusers/pipelines/omnigen/pipeline_omnigen.py index 5fe5be3b26d2..ea86d6c638fa 100644 --- a/src/diffusers/pipelines/omnigen/pipeline_omnigen.py +++ b/src/diffusers/pipelines/omnigen/pipeline_omnigen.py @@ -146,6 +146,9 @@ def __init__( tokenizer: LlamaTokenizer, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -406,7 +409,6 @@ def __call__( ) self._guidance_scale = guidance_scale - self._interrupt = False # 2. Define call parameters batch_size = len(prompt) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py index bc90073cba77..e87e15ab30a0 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py @@ -228,6 +228,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py index 6d89f16765a3..598dd1da154f 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py @@ -205,6 +205,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py index 83540885bfb2..c7de3625b9c4 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py @@ -274,6 +274,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["down.block_2", "up.block_1.attentions_0"], "mid" ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py index b84f5d555914..76454a952c4c 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py @@ -264,6 +264,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py index a6a8deb5883c..c2860017f5e4 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py @@ -216,6 +216,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "blocks.1", # "blocks.16.attn1", "blocks.16", "16", 16 ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -736,7 +740,6 @@ def __call__( ) self._guidance_scale = guidance_scale self._guidance_rescale = guidance_rescale - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_kolors.py b/src/diffusers/pipelines/pag/pipeline_pag_kolors.py index 62f634312ada..f666246301d4 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_kolors.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_kolors.py @@ -191,6 +191,11 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -883,7 +888,6 @@ def __call__( self._guidance_scale = guidance_scale self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sana.py b/src/diffusers/pipelines/pag/pipeline_pag_sana.py index a233f70136af..67951ed1a5f5 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sana.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sana.py @@ -168,6 +168,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "transformer_blocks.0", ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -786,7 +789,6 @@ def __call__( self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale self._guidance_scale = guidance_scale - self._interrupt = False # 2. Default height and width to transformer if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd.py b/src/diffusers/pipelines/pag/pipeline_pag_sd.py index fc7dc3a83f27..5dc02cd71c86 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd.py @@ -215,6 +215,11 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( @@ -887,7 +892,6 @@ def __call__( self._guidance_rescale = guidance_rescale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py index fde3e500a573..0c10ac4599b4 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py @@ -188,6 +188,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "blocks.1", # 1st transformer block ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -835,7 +839,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale # diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py index d64582a26f7a..a2b16e63d0f5 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py @@ -204,6 +204,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "blocks.1", # 1st transformer block ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -895,7 +899,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py index d3a015e569c1..c0186645e25a 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py @@ -143,6 +143,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid_block.*attn1", # ["mid"], ["down_blocks.1"] ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py index d91c02b607a3..4006d8e90d3b 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py @@ -210,6 +210,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( @@ -919,7 +923,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py index db652989cfc1..ef30bad888cb 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py @@ -242,6 +242,11 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( @@ -1062,7 +1067,6 @@ def __call__( self._guidance_rescale = guidance_rescale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py index 856f6a3e789e..fb86c3995ab8 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py @@ -262,6 +262,12 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"],["down.block_1"],["up.block_0.attentions_0"] ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1067,7 +1073,6 @@ def __call__( self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py index 93dcca0ea9d6..4107d1b4e940 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py @@ -284,6 +284,13 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._denoising_start=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1237,7 +1244,6 @@ def __call__( self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end self._denoising_start = denoising_start - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py index 8b06bdc9c969..259f028ec9cd 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py @@ -300,6 +300,13 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._denoising_start=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1360,7 +1367,6 @@ def __call__( self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end self._denoising_start = denoising_start - self._interrupt = False self._pag_scale = pag_scale self._pag_adaptive_scale = pag_adaptive_scale diff --git a/src/diffusers/pipelines/pia/pipeline_pia.py b/src/diffusers/pipelines/pia/pipeline_pia.py index df8499ab900a..658dad24b3a8 100644 --- a/src/diffusers/pipelines/pia/pipeline_pia.py +++ b/src/diffusers/pipelines/pia/pipeline_pia.py @@ -191,6 +191,9 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/sana/pipeline_sana.py b/src/diffusers/pipelines/sana/pipeline_sana.py index 34b84a89e60d..7233605e3340 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana.py +++ b/src/diffusers/pipelines/sana/pipeline_sana.py @@ -207,6 +207,10 @@ def __init__( scheduler: DPMSolverMultistepScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -863,7 +867,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False # 2. Default height and width to transformer if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py b/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py index a7c7d027fbda..78d3e96665c1 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py +++ b/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py @@ -215,6 +215,10 @@ def __init__( scheduler: DPMSolverMultistepScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( tokenizer=tokenizer, @@ -926,7 +930,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False # 2. Default height and width to transformer if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/sana/pipeline_sana_sprint.py b/src/diffusers/pipelines/sana/pipeline_sana_sprint.py index 03b306b539c0..a1e6248947ef 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana_sprint.py +++ b/src/diffusers/pipelines/sana/pipeline_sana_sprint.py @@ -158,6 +158,10 @@ def __init__( scheduler: DPMSolverMultistepScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler @@ -735,7 +739,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._interrupt = False # 2. Default height and width to transformer if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py index 38f1c4314e4f..c8454ccd1033 100644 --- a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py +++ b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py @@ -99,6 +99,8 @@ def __init__( latent_dim_scale: float = 10.67, ) -> None: super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( decoder=decoder, tokenizer=tokenizer, diff --git a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py index 241c454e103e..16a426554cfe 100644 --- a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py +++ b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py @@ -120,6 +120,8 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ) -> None: super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py index 6e93c34929de..da0512b2c727 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py @@ -210,6 +210,11 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( @@ -944,7 +949,6 @@ def __call__( self._guidance_rescale = guidance_rescale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py index f158c41cac53..edee14355c28 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py @@ -129,6 +129,9 @@ def __init__( feature_extractor: DPTImageProcessor, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 is_unet_version_less_0_9_0 = ( unet is not None diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py index 901dcd6db012..aa4e8900ec46 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py @@ -238,6 +238,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( @@ -1004,7 +1008,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py index 0f7be1a1bbcd..3f6c9cfbf61f 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py @@ -184,6 +184,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( @@ -1093,7 +1097,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py index e0748943ffff..eefc613b2bfb 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py @@ -138,6 +138,8 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py index 4618d384cbd7..e0076d3536dc 100644 --- a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +++ b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py @@ -201,6 +201,10 @@ def __init__( feature_extractor: SiglipImageProcessor = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -945,7 +949,6 @@ def __call__( self._skip_layer_guidance_scale = skip_layer_guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py index 19bdc9792e23..8886510f34a6 100644 --- a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py @@ -222,6 +222,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -984,7 +988,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py index cac305a87f00..55b788328285 100644 --- a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py @@ -221,6 +221,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() + self._guidance_scale=1.0 + self._joint_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1112,7 +1116,6 @@ def __call__( self._guidance_scale = guidance_scale self._clip_skip = clip_skip self._joint_attention_kwargs = joint_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py index c7c5bd9cff67..8e15f50f17f6 100644 --- a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py @@ -157,6 +157,7 @@ def __init__( force_zeros_for_empty_prompt: bool = True, ): super().__init__() + self._guidance_scale=1.0 # get correct sigmas from LMS scheduler = LMSDiscreteScheduler.from_config(scheduler.config) diff --git a/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py b/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py index 702f3eda5816..6ec23b72c821 100644 --- a/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +++ b/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py @@ -236,6 +236,11 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if safety_checker is None and requires_safety_checker: logger.warning( @@ -889,7 +894,6 @@ def __call__( self._guidance_rescale = guidance_rescale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py b/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py index ccee6d47b47a..9f001f46106b 100644 --- a/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +++ b/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py @@ -212,6 +212,11 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._num_timesteps=0 + self._interrupt=False if safety_checker is None and requires_safety_checker: logger.warning( @@ -950,7 +955,6 @@ def __call__( self._guidance_rescale = guidance_rescale self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py index 9c69fe65fbdb..6f8157026352 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py @@ -256,6 +256,12 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1069,7 +1075,6 @@ def __call__( self._clip_skip = clip_skip self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py index 08d0b44d613d..e5878c51afc7 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py @@ -277,6 +277,13 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._denoising_start=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1232,7 +1239,6 @@ def __call__( self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end self._denoising_start = denoising_start - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py index 835c0af800da..6c598eb7d3c6 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py @@ -307,6 +307,13 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() + self._guidance_scale=1.0 + self._guidance_rescale=0.0 + self._cross_attention_kwargs=None + self._denoising_end=None + self._denoising_start=None + self._num_timesteps=0 + self._interrupt=False self.register_modules( vae=vae, @@ -1370,7 +1377,6 @@ def __call__( self._cross_attention_kwargs = cross_attention_kwargs self._denoising_end = denoising_end self._denoising_start = denoising_start - self._interrupt = False # 2. Define call parameters if prompt is not None and isinstance(prompt, str): diff --git a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py index 8c1af7863e63..139d2f130635 100644 --- a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +++ b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py @@ -177,6 +177,8 @@ def __init__( feature_extractor: CLIPImageProcessor, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( vae=vae, 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 6cd0e415e129..8d6e8a3363ac 100644 --- a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +++ b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py @@ -240,6 +240,7 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() + self._guidance_scale=1.0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py index 5eacb64d01e3..68a51f2d8d81 100644 --- a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +++ b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py @@ -288,6 +288,7 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() + self._guidance_scale=1.0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/wan/pipeline_wan.py b/src/diffusers/pipelines/wan/pipeline_wan.py index 572995530fed..1a11ce2e32cb 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan.py +++ b/src/diffusers/pipelines/wan/pipeline_wan.py @@ -126,6 +126,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._interrupt=False + self._attention_kwargs=None self.register_modules( vae=vae, @@ -469,8 +474,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False device = self._execution_device diff --git a/src/diffusers/pipelines/wan/pipeline_wan_i2v.py b/src/diffusers/pipelines/wan/pipeline_wan_i2v.py index b3e1222a46c6..b7c3f4eadcf8 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_i2v.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_i2v.py @@ -165,6 +165,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._interrupt=False + self._attention_kwargs=None self.register_modules( vae=vae, @@ -602,8 +607,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False device = self._execution_device diff --git a/src/diffusers/pipelines/wan/pipeline_wan_video2video.py b/src/diffusers/pipelines/wan/pipeline_wan_video2video.py index b2b151623939..e888ff13e549 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_video2video.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_video2video.py @@ -205,6 +205,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 + self._current_timestep=None + self._interrupt=False + self._attention_kwargs=None self.register_modules( vae=vae, @@ -593,8 +598,6 @@ def __call__( self._guidance_scale = guidance_scale self._attention_kwargs = attention_kwargs - self._current_timestep = None - self._interrupt = False device = self._execution_device diff --git a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py index edc01f0d5c75..99e3c073282b 100644 --- a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py +++ b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py @@ -98,6 +98,8 @@ def __init__( latent_dim_scale: float = 10.67, ) -> None: super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, diff --git a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py index 8f6ba419721d..4cacb32472ec 100644 --- a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +++ b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py @@ -116,6 +116,8 @@ def __init__( resolution_multiple: float = 42.67, ) -> None: super().__init__() + self._guidance_scale=1.0 + self._num_timesteps=0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, From 8336e44d3e948ce84b50f904c2b0f0e544172c59 Mon Sep 17 00:00:00 2001 From: ishan-modi Date: Mon, 26 May 2025 13:19:36 +0530 Subject: [PATCH 2/2] fix style --- .../pipelines/allegro/pipeline_allegro.py | 8 ++++---- .../pipelines/animatediff/pipeline_animatediff.py | 8 ++++---- .../animatediff/pipeline_animatediff_controlnet.py | 8 ++++---- .../animatediff/pipeline_animatediff_sdxl.py | 12 ++++++------ .../animatediff/pipeline_animatediff_sparsectrl.py | 6 +++--- .../pipeline_animatediff_video2video.py | 8 ++++---- .../pipeline_animatediff_video2video_controlnet.py | 8 ++++---- .../pipelines/aura_flow/pipeline_aura_flow.py | 6 +++--- .../pipelines/cogvideo/pipeline_cogvideox.py | 10 +++++----- .../cogvideo/pipeline_cogvideox_fun_control.py | 10 +++++----- .../cogvideo/pipeline_cogvideox_image2video.py | 10 +++++----- .../cogvideo/pipeline_cogvideox_video2video.py | 10 +++++----- .../pipelines/cogview3/pipeline_cogview3plus.py | 6 +++--- .../pipelines/cogview4/pipeline_cogview4.py | 10 +++++----- .../cogview4/pipeline_cogview4_control.py | 10 +++++----- .../pipelines/consisid/pipeline_consisid.py | 8 ++++---- .../pipelines/controlnet/pipeline_controlnet.py | 8 ++++---- .../controlnet/pipeline_controlnet_img2img.py | 8 ++++---- .../controlnet/pipeline_controlnet_inpaint.py | 8 ++++---- .../pipeline_controlnet_inpaint_sd_xl.py | 8 ++++---- .../controlnet/pipeline_controlnet_sd_xl.py | 10 +++++----- .../pipeline_controlnet_sd_xl_img2img.py | 8 ++++---- .../pipeline_controlnet_union_inpaint_sd_xl.py | 8 ++++---- .../controlnet/pipeline_controlnet_union_sd_xl.py | 10 +++++----- .../pipeline_controlnet_union_sd_xl_img2img.py | 8 ++++---- .../pipeline_hunyuandit_controlnet.py | 8 ++++---- .../pipeline_stable_diffusion_3_controlnet.py | 8 ++++---- ...ine_stable_diffusion_3_controlnet_inpainting.py | 8 ++++---- .../controlnet_xs/pipeline_controlnet_xs.py | 6 +++--- .../controlnet_xs/pipeline_controlnet_xs_sd_xl.py | 6 +++--- .../alt_diffusion/pipeline_alt_diffusion.py | 8 ++++---- .../pipeline_alt_diffusion_img2img.py | 6 +++--- .../pipelines/easyanimate/pipeline_easyanimate.py | 8 ++++---- .../easyanimate/pipeline_easyanimate_control.py | 8 ++++---- .../easyanimate/pipeline_easyanimate_inpaint.py | 8 ++++---- src/diffusers/pipelines/flux/pipeline_flux.py | 10 +++++----- .../pipelines/flux/pipeline_flux_control.py | 8 ++++---- .../flux/pipeline_flux_control_img2img.py | 8 ++++---- .../flux/pipeline_flux_control_inpaint.py | 8 ++++---- .../pipelines/flux/pipeline_flux_controlnet.py | 8 ++++---- .../pipeline_flux_controlnet_image_to_image.py | 8 ++++---- .../flux/pipeline_flux_controlnet_inpainting.py | 8 ++++---- src/diffusers/pipelines/flux/pipeline_flux_fill.py | 8 ++++---- .../pipelines/flux/pipeline_flux_img2img.py | 8 ++++---- .../pipelines/flux/pipeline_flux_inpaint.py | 8 ++++---- .../hidream_image/pipeline_hidream_image.py | 8 ++++---- .../pipeline_hunyuan_skyreels_image2video.py | 10 +++++----- .../hunyuan_video/pipeline_hunyuan_video.py | 10 +++++----- .../pipeline_hunyuan_video_image2video.py | 10 +++++----- .../pipelines/hunyuandit/pipeline_hunyuandit.py | 8 ++++---- .../pipelines/i2vgen_xl/pipeline_i2vgen_xl.py | 2 +- .../kandinsky2_2/pipeline_kandinsky2_2.py | 4 ++-- .../kandinsky2_2/pipeline_kandinsky2_2_img2img.py | 4 ++-- .../pipeline_kandinsky2_2_inpainting.py | 4 ++-- .../kandinsky2_2/pipeline_kandinsky2_2_prior.py | 4 ++-- .../pipelines/kandinsky3/pipeline_kandinsky3.py | 4 ++-- .../kandinsky3/pipeline_kandinsky3_img2img.py | 4 ++-- src/diffusers/pipelines/kolors/pipeline_kolors.py | 10 +++++----- .../pipelines/kolors/pipeline_kolors_img2img.py | 12 ++++++------ .../pipeline_latent_consistency_img2img.py | 6 +++--- .../pipeline_latent_consistency_text2img.py | 6 +++--- src/diffusers/pipelines/latte/pipeline_latte.py | 8 ++++---- .../pipeline_leditspp_stable_diffusion.py | 4 ++-- .../pipeline_leditspp_stable_diffusion_xl.py | 10 +++++----- src/diffusers/pipelines/ltx/pipeline_ltx.py | 10 +++++----- .../pipelines/ltx/pipeline_ltx_condition.py | 10 +++++----- .../pipelines/ltx/pipeline_ltx_image2video.py | 10 +++++----- src/diffusers/pipelines/lumina/pipeline_lumina.py | 4 ++-- .../pipelines/lumina2/pipeline_lumina2.py | 6 +++--- src/diffusers/pipelines/mochi/pipeline_mochi.py | 10 +++++----- .../pipelines/omnigen/pipeline_omnigen.py | 6 +++--- .../pipelines/pag/pipeline_pag_controlnet_sd.py | 6 +++--- .../pag/pipeline_pag_controlnet_sd_inpaint.py | 6 +++--- .../pipelines/pag/pipeline_pag_controlnet_sd_xl.py | 8 ++++---- .../pag/pipeline_pag_controlnet_sd_xl_img2img.py | 6 +++--- .../pipelines/pag/pipeline_pag_hunyuandit.py | 8 ++++---- src/diffusers/pipelines/pag/pipeline_pag_kolors.py | 10 +++++----- src/diffusers/pipelines/pag/pipeline_pag_sana.py | 6 +++--- src/diffusers/pipelines/pag/pipeline_pag_sd.py | 10 +++++----- src/diffusers/pipelines/pag/pipeline_pag_sd_3.py | 8 ++++---- .../pipelines/pag/pipeline_pag_sd_3_img2img.py | 8 ++++---- .../pipelines/pag/pipeline_pag_sd_animatediff.py | 6 +++--- .../pipelines/pag/pipeline_pag_sd_img2img.py | 8 ++++---- .../pipelines/pag/pipeline_pag_sd_inpaint.py | 10 +++++----- src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py | 12 ++++++------ .../pipelines/pag/pipeline_pag_sd_xl_img2img.py | 14 +++++++------- .../pipelines/pag/pipeline_pag_sd_xl_inpaint.py | 14 +++++++------- src/diffusers/pipelines/pia/pipeline_pia.py | 6 +++--- src/diffusers/pipelines/sana/pipeline_sana.py | 8 ++++---- .../pipelines/sana/pipeline_sana_controlnet.py | 8 ++++---- .../pipelines/sana/pipeline_sana_sprint.py | 8 ++++---- .../stable_cascade/pipeline_stable_cascade.py | 4 ++-- .../pipeline_stable_cascade_prior.py | 4 ++-- .../stable_diffusion/pipeline_stable_diffusion.py | 10 +++++----- .../pipeline_stable_diffusion_depth2img.py | 6 +++--- .../pipeline_stable_diffusion_img2img.py | 8 ++++---- .../pipeline_stable_diffusion_inpaint.py | 8 ++++---- .../pipeline_stable_diffusion_instruct_pix2pix.py | 4 ++-- .../pipeline_stable_diffusion_3.py | 8 ++++---- .../pipeline_stable_diffusion_3_img2img.py | 8 ++++---- .../pipeline_stable_diffusion_3_inpaint.py | 8 ++++---- .../pipeline_stable_diffusion_xl_k_diffusion.py | 2 +- .../pipeline_stable_diffusion_ldm3d.py | 10 +++++----- .../pipeline_stable_diffusion_panorama.py | 10 +++++----- .../pipeline_stable_diffusion_xl.py | 12 ++++++------ .../pipeline_stable_diffusion_xl_img2img.py | 14 +++++++------- .../pipeline_stable_diffusion_xl_inpaint.py | 14 +++++++------- .../pipeline_stable_video_diffusion.py | 4 ++-- .../pipeline_stable_diffusion_adapter.py | 2 +- .../pipeline_stable_diffusion_xl_adapter.py | 2 +- src/diffusers/pipelines/wan/pipeline_wan.py | 10 +++++----- src/diffusers/pipelines/wan/pipeline_wan_i2v.py | 10 +++++----- .../pipelines/wan/pipeline_wan_video2video.py | 10 +++++----- .../pipelines/wuerstchen/pipeline_wuerstchen.py | 4 ++-- .../wuerstchen/pipeline_wuerstchen_prior.py | 4 ++-- 115 files changed, 450 insertions(+), 450 deletions(-) diff --git a/src/diffusers/pipelines/allegro/pipeline_allegro.py b/src/diffusers/pipelines/allegro/pipeline_allegro.py index 5c368dc2b23a..2a130322c36b 100644 --- a/src/diffusers/pipelines/allegro/pipeline_allegro.py +++ b/src/diffusers/pipelines/allegro/pipeline_allegro.py @@ -198,10 +198,10 @@ def __init__( scheduler: KarrasDiffusionSchedulers, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff.py index da9c418212a2..0d0d7749e786 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff.py @@ -136,10 +136,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py index 97aef02544a0..40a0343733c1 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py @@ -172,10 +172,10 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py index 7ede5b454c0d..5563dc3e817a 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py @@ -299,12 +299,12 @@ def __init__( force_zeros_for_empty_prompt: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py index a1c4eb9605a3..3df525187ac4 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py @@ -184,9 +184,9 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py index 17edd1e483f5..e06eac445515 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py @@ -239,10 +239,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py index 22d42662060f..47deeddaf7f1 100644 --- a/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py +++ b/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py @@ -262,10 +262,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py b/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py index 3edbfa81c804..1a347235a54c 100644 --- a/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py +++ b/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py @@ -154,9 +154,9 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._attention_kwargs = None + self._num_timesteps = 0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py index 39b9864af3c3..a8964f370e81 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py @@ -185,11 +185,11 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py index 533521d43783..0825150ad718 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py @@ -192,11 +192,11 @@ def __init__( scheduler: KarrasDiffusionSchedulers, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py index 0816e3f334b5..790eba32c687 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py @@ -202,11 +202,11 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, diff --git a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py index 2024fe4dd8d4..2bd2e42e6e55 100644 --- a/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py +++ b/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py @@ -207,11 +207,11 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py index 8aa690a5a017..99de88a5370f 100644 --- a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py +++ b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py @@ -156,9 +156,9 @@ def __init__( scheduler: Union[CogVideoXDDIMScheduler, CogVideoXDPMScheduler], ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py index 4205a1876b4d..fde03c953728 100644 --- a/src/diffusers/pipelines/cogview4/pipeline_cogview4.py +++ b/src/diffusers/pipelines/cogview4/pipeline_cogview4.py @@ -168,11 +168,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py b/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py index a352c0e05f68..a0b0c9496ef9 100644 --- a/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py +++ b/src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py @@ -170,11 +170,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/consisid/pipeline_consisid.py b/src/diffusers/pipelines/consisid/pipeline_consisid.py index e09e11f38fa9..4b3e75d532b7 100644 --- a/src/diffusers/pipelines/consisid/pipeline_consisid.py +++ b/src/diffusers/pipelines/consisid/pipeline_consisid.py @@ -288,10 +288,10 @@ def __init__( scheduler: CogVideoXDPMScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet.py index e8e6c04c2552..bdc3230b786d 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet.py @@ -223,10 +223,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py index eb7f5c6bc356..7009f2959b12 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py @@ -201,10 +201,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py index 4a89dfb22709..a194e968e28f 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py @@ -207,10 +207,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py index b55163a746a2..946668c32d63 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py @@ -257,10 +257,10 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py index 691243ab6624..4602d2160a70 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py @@ -269,11 +269,11 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py index 5a6d1c56221a..9077b3a4ea47 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py @@ -262,10 +262,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py index 2e83c1b02209..b7495aec2031 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py @@ -239,10 +239,10 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if not isinstance(controlnet, ControlNetUnionModel): raise ValueError("Expected `controlnet` to be of type `ControlNetUnionModel`.") diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py index 4bbe071ea396..465ddb293e01 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py @@ -259,11 +259,11 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetUnionModel(controlnet) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py index bdef279be33b..51c4e31ead4d 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py @@ -271,10 +271,10 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if not isinstance(controlnet, ControlNetUnionModel): raise ValueError("Expected `controlnet` to be of type `ControlNetUnionModel`.") diff --git a/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py b/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py index 494172f7119c..3d683d30b2b9 100644 --- a/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py +++ b/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py @@ -237,10 +237,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = HunyuanDiT2DMultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py index 9c05a2793ac4..403127e468f8 100644 --- a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +++ b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py @@ -206,10 +206,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = SD3MultiControlNetModel(controlnet) if isinstance(controlnet, SD3MultiControlNetModel): diff --git a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py index e0c737ff5e5d..4164c679029f 100644 --- a/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py +++ b/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py @@ -227,10 +227,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py index 8776f679a988..0a238a11d755 100644 --- a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +++ b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py @@ -156,9 +156,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if isinstance(unet, UNet2DConditionModel): unet = UNetControlNetXSModel.from_unet(unet, controlnet) diff --git a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py index e2bd56990792..706dfe3ac917 100644 --- a/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +++ b/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py @@ -191,9 +191,9 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if isinstance(unet, UNet2DConditionModel): unet = UNetControlNetXSModel.from_unet(unet, controlnet) diff --git a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py index 1e073d98e162..1d40ff5b56d2 100644 --- a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +++ b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py @@ -209,10 +209,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py index 8f474c2874bd..b66bba214076 100644 --- a/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +++ b/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py @@ -237,9 +237,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py index f0d70ff445f1..05ba713ad8ae 100755 --- a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py +++ b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate.py @@ -217,10 +217,10 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py index 65dbb27cd9b1..2817d4cf0c79 100755 --- a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py +++ b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_control.py @@ -319,10 +319,10 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py index 671e463095a4..f4377193bc7f 100755 --- a/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py +++ b/src/diffusers/pipelines/easyanimate/pipeline_easyanimate_inpaint.py @@ -361,10 +361,10 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux.py b/src/diffusers/pipelines/flux/pipeline_flux.py index d6b39beb42d0..cb16853720d7 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux.py +++ b/src/diffusers/pipelines/flux/pipeline_flux.py @@ -193,11 +193,11 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._current_timestep = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control.py b/src/diffusers/pipelines/flux/pipeline_flux_control.py index 6d39491f9403..56008a8e3f25 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_control.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_control.py @@ -203,10 +203,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py b/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py index cb3c45af4681..def910804787 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py @@ -217,10 +217,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py b/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py index 6d5ce5391605..1d65d6d6edfe 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py @@ -248,10 +248,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py index 4482c22c679c..0961d92eee68 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py @@ -221,10 +221,10 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = FluxMultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py index 250d6f3e5a86..891def4b0ab9 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py @@ -214,10 +214,10 @@ def __init__( ], ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = FluxMultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py index 153671b1562a..b5e377691ddb 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py @@ -216,10 +216,10 @@ def __init__( ], ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if isinstance(controlnet, (list, tuple)): controlnet = FluxMultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_fill.py b/src/diffusers/pipelines/flux/pipeline_flux_fill.py index f0c8dbf6fb35..5d20a92cf395 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_fill.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_fill.py @@ -211,10 +211,10 @@ def __init__( transformer: FluxTransformer2DModel, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_img2img.py b/src/diffusers/pipelines/flux/pipeline_flux_img2img.py index 78499a074af0..34f3195639ae 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_img2img.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_img2img.py @@ -210,10 +210,10 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py b/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py index 3a49cb5e3fbb..f03679f5d762 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py @@ -207,10 +207,10 @@ def __init__( feature_extractor: CLIPImageProcessor = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py b/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py index bc6faa491bea..1d26e25ae7b7 100644 --- a/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py +++ b/src/diffusers/pipelines/hidream_image/pipeline_hidream_image.py @@ -162,10 +162,10 @@ def __init__( transformer: HiDreamImageTransformer2DModel, ): super().__init__() - self._guidance_scale=1.0 - self._attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py index a45bacb003d0..083062dc887a 100644 --- a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py +++ b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_skyreels_image2video.py @@ -206,11 +206,11 @@ def __init__( tokenizer_2: CLIPTokenizer, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py index c14dc3d3b392..0e2747fe0399 100644 --- a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py +++ b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py @@ -181,11 +181,11 @@ def __init__( tokenizer_2: CLIPTokenizer, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py index 39caccb255b9..59c98d0a63b1 100644 --- a/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py +++ b/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py @@ -259,11 +259,11 @@ def __init__( image_processor: CLIPImageProcessor, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py b/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py index 4c98bd5de047..a7c168534b5e 100644 --- a/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py +++ b/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py @@ -211,10 +211,10 @@ def __init__( tokenizer_2: Optional[MT5Tokenizer] = None, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py b/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py index c941463ea26c..88d1e5dd0259 100644 --- a/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py +++ b/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py @@ -132,7 +132,7 @@ def __init__( scheduler: DDIMScheduler, ): super().__init__() - self._guidance_scale=1.0 + self._guidance_scale = 1.0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py index 42421716c4b8..2eaa0608c532 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py @@ -95,8 +95,8 @@ def __init__( movq: VQModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( unet=unet, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py index 9bd3cf4534f8..6b4ca0e08376 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py @@ -101,8 +101,8 @@ def __init__( movq: VQModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( unet=unet, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py index 0d532d367604..2047bc11fff7 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py @@ -266,8 +266,8 @@ def __init__( movq: VQModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( unet=unet, diff --git a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py index 3207762ff59d..f4e4ab5335d9 100644 --- a/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py +++ b/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py @@ -127,8 +127,8 @@ def __init__( image_processor: CLIPImageProcessor, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( prior=prior, diff --git a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py index dc7347b64ab9..338505f13047 100644 --- a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +++ b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py @@ -75,8 +75,8 @@ def __init__( movq: VQModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, unet=unet, scheduler=scheduler, movq=movq diff --git a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py index 20af7f9e886a..124c3db95138 100644 --- a/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +++ b/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py @@ -72,8 +72,8 @@ def __init__( movq: VQModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, unet=unet, scheduler=scheduler, movq=movq diff --git a/src/diffusers/pipelines/kolors/pipeline_kolors.py b/src/diffusers/pipelines/kolors/pipeline_kolors.py index 581ae51ee419..1e4d977f84e5 100644 --- a/src/diffusers/pipelines/kolors/pipeline_kolors.py +++ b/src/diffusers/pipelines/kolors/pipeline_kolors.py @@ -177,11 +177,11 @@ def __init__( force_zeros_for_empty_prompt: bool = False, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py b/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py index c8cbfd033338..4efa9f9290c8 100644 --- a/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py +++ b/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py @@ -196,12 +196,12 @@ def __init__( force_zeros_for_empty_prompt: bool = False, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._denoising_start=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._denoising_start = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py index d5ae98fcbb14..c7c63874338e 100644 --- a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +++ b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py @@ -212,9 +212,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py index 46195876298a..1c36bc45e68e 100644 --- a/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +++ b/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py @@ -191,9 +191,9 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/latte/pipeline_latte.py b/src/diffusers/pipelines/latte/pipeline_latte.py index 67d01c9663ca..af594b16e896 100644 --- a/src/diffusers/pipelines/latte/pipeline_latte.py +++ b/src/diffusers/pipelines/latte/pipeline_latte.py @@ -185,10 +185,10 @@ def __init__( scheduler: KarrasDiffusionSchedulers, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py index 1d122f1bc878..0ce73c21284c 100644 --- a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +++ b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py @@ -315,8 +315,8 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None if not isinstance(scheduler, DDIMScheduler) and not isinstance(scheduler, DPMSolverMultistepScheduler): scheduler = DPMSolverMultistepScheduler.from_config( diff --git a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py index bd987b82e944..85973b8eecb8 100644 --- a/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +++ b/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py @@ -359,11 +359,11 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx.py b/src/diffusers/pipelines/ltx/pipeline_ltx.py index 6dd1d6e9a51e..5e921a426c0b 100644 --- a/src/diffusers/pipelines/ltx/pipeline_ltx.py +++ b/src/diffusers/pipelines/ltx/pipeline_ltx.py @@ -177,11 +177,11 @@ def __init__( transformer: LTXVideoTransformer3DModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._attention_kwargs=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._attention_kwargs = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py b/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py index a2a50085794b..f0ba5987e0ef 100644 --- a/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py +++ b/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py @@ -259,11 +259,11 @@ def __init__( transformer: LTXVideoTransformer3DModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._attention_kwargs=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._attention_kwargs = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py b/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py index 51b1b9a5c3f0..d5452897bac1 100644 --- a/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py +++ b/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py @@ -196,11 +196,11 @@ def __init__( transformer: LTXVideoTransformer3DModel, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._attention_kwargs=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._attention_kwargs = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/lumina/pipeline_lumina.py b/src/diffusers/pipelines/lumina/pipeline_lumina.py index cff89812db3f..cb2a32bc474a 100644 --- a/src/diffusers/pipelines/lumina/pipeline_lumina.py +++ b/src/diffusers/pipelines/lumina/pipeline_lumina.py @@ -185,8 +185,8 @@ def __init__( tokenizer: Union[GemmaTokenizer, GemmaTokenizerFast], ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/lumina2/pipeline_lumina2.py b/src/diffusers/pipelines/lumina2/pipeline_lumina2.py index 54ba1c6f64b9..b65088f910f1 100644 --- a/src/diffusers/pipelines/lumina2/pipeline_lumina2.py +++ b/src/diffusers/pipelines/lumina2/pipeline_lumina2.py @@ -167,9 +167,9 @@ def __init__( tokenizer: Union[GemmaTokenizer, GemmaTokenizerFast], ): super().__init__() - self._guidance_scale=1.0 - self._attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._attention_kwargs = None + self._num_timesteps = 0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/mochi/pipeline_mochi.py b/src/diffusers/pipelines/mochi/pipeline_mochi.py index d368c3d0cb66..1f42828df95b 100644 --- a/src/diffusers/pipelines/mochi/pipeline_mochi.py +++ b/src/diffusers/pipelines/mochi/pipeline_mochi.py @@ -177,11 +177,11 @@ def __init__( force_zeros_for_empty_prompt: bool = False, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._attention_kwargs=None - self._current_timestep=None - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._attention_kwargs = None + self._current_timestep = None + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/omnigen/pipeline_omnigen.py b/src/diffusers/pipelines/omnigen/pipeline_omnigen.py index ea86d6c638fa..c6d496697f7f 100644 --- a/src/diffusers/pipelines/omnigen/pipeline_omnigen.py +++ b/src/diffusers/pipelines/omnigen/pipeline_omnigen.py @@ -146,9 +146,9 @@ def __init__( tokenizer: LlamaTokenizer, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py index e87e15ab30a0..5e1e74744ee6 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py @@ -228,9 +228,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py index 598dd1da154f..e1a1c5da33ee 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py @@ -205,9 +205,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py index c7de3625b9c4..91dd99bfa57d 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py @@ -274,10 +274,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["down.block_2", "up.block_1.attentions_0"], "mid" ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py index 76454a952c4c..ffc53985c782 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py @@ -264,9 +264,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if isinstance(controlnet, (list, tuple)): controlnet = MultiControlNetModel(controlnet) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py index c2860017f5e4..5326292bbdbe 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py @@ -216,10 +216,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "blocks.1", # "blocks.16.attn1", "blocks.16", "16", 16 ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_kolors.py b/src/diffusers/pipelines/pag/pipeline_pag_kolors.py index f666246301d4..229218dacad1 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_kolors.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_kolors.py @@ -191,11 +191,11 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sana.py b/src/diffusers/pipelines/pag/pipeline_pag_sana.py index 67951ed1a5f5..7967cf975bea 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sana.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sana.py @@ -168,9 +168,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "transformer_blocks.0", ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd.py b/src/diffusers/pipelines/pag/pipeline_pag_sd.py index 5dc02cd71c86..d5aa8b077d74 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd.py @@ -215,11 +215,11 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py index 0c10ac4599b4..e41ae917c212 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py @@ -188,10 +188,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "blocks.1", # 1st transformer block ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py index a2b16e63d0f5..6028827e4e75 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py @@ -204,10 +204,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "blocks.1", # 1st transformer block ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py index c0186645e25a..74ca47da8dc2 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py @@ -143,9 +143,9 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid_block.*attn1", # ["mid"], ["down_blocks.1"] ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py index 4006d8e90d3b..2a1608ef3775 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py @@ -210,10 +210,10 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py index ef30bad888cb..f231ab2b228a 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py @@ -242,11 +242,11 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py index fb86c3995ab8..0a07bdf3eced 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py @@ -262,12 +262,12 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"],["down.block_1"],["up.block_0.attentions_0"] ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py index 4107d1b4e940..3a62988b82a5 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py @@ -284,13 +284,13 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._denoising_start=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._denoising_start = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py index 259f028ec9cd..2e694119447a 100644 --- a/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +++ b/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py @@ -300,13 +300,13 @@ def __init__( pag_applied_layers: Union[str, List[str]] = "mid", # ["mid"], ["down.block_1", "up.block_0.attentions_0"] ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._denoising_start=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._denoising_start = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/pia/pipeline_pia.py b/src/diffusers/pipelines/pia/pipeline_pia.py index 658dad24b3a8..3eefa99f4287 100644 --- a/src/diffusers/pipelines/pia/pipeline_pia.py +++ b/src/diffusers/pipelines/pia/pipeline_pia.py @@ -191,9 +191,9 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 if isinstance(unet, UNet2DConditionModel): unet = UNetMotionModel.from_unet2d(unet, motion_adapter) diff --git a/src/diffusers/pipelines/sana/pipeline_sana.py b/src/diffusers/pipelines/sana/pipeline_sana.py index 7233605e3340..c7cc2e733948 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana.py +++ b/src/diffusers/pipelines/sana/pipeline_sana.py @@ -207,10 +207,10 @@ def __init__( scheduler: DPMSolverMultistepScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py b/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py index 78d3e96665c1..096d7e27933f 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py +++ b/src/diffusers/pipelines/sana/pipeline_sana_controlnet.py @@ -215,10 +215,10 @@ def __init__( scheduler: DPMSolverMultistepScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( tokenizer=tokenizer, diff --git a/src/diffusers/pipelines/sana/pipeline_sana_sprint.py b/src/diffusers/pipelines/sana/pipeline_sana_sprint.py index a1e6248947ef..d5b139035aec 100644 --- a/src/diffusers/pipelines/sana/pipeline_sana_sprint.py +++ b/src/diffusers/pipelines/sana/pipeline_sana_sprint.py @@ -158,10 +158,10 @@ def __init__( scheduler: DPMSolverMultistepScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler diff --git a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py index c8454ccd1033..f1383c8e463c 100644 --- a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py +++ b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py @@ -99,8 +99,8 @@ def __init__( latent_dim_scale: float = 10.67, ) -> None: super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( decoder=decoder, tokenizer=tokenizer, diff --git a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py index 16a426554cfe..2f81b7b9407f 100644 --- a/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py +++ b/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py @@ -120,8 +120,8 @@ def __init__( image_encoder: Optional[CLIPVisionModelWithProjection] = None, ) -> None: super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py index da0512b2c727..10ff9bbf02a5 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py @@ -210,11 +210,11 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py index edee14355c28..c0ec00a5fe95 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py @@ -129,9 +129,9 @@ def __init__( feature_extractor: DPTImageProcessor, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 is_unet_version_less_0_9_0 = ( unet is not None diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py index aa4e8900ec46..e00949e305cf 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py @@ -238,10 +238,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py index 3f6c9cfbf61f..90d9bbf74d6a 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py @@ -184,10 +184,10 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if scheduler is not None and getattr(scheduler.config, "steps_offset", 1) != 1: deprecation_message = ( diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py index eefc613b2bfb..825781f242b3 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py @@ -138,8 +138,8 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py index e0076d3536dc..233f6a9374e1 100644 --- a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +++ b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py @@ -201,10 +201,10 @@ def __init__( feature_extractor: SiglipImageProcessor = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py index 8886510f34a6..37a776480e26 100644 --- a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py @@ -222,10 +222,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py index 55b788328285..81d38c45308e 100644 --- a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py @@ -221,10 +221,10 @@ def __init__( feature_extractor: Optional[SiglipImageProcessor] = None, ): super().__init__() - self._guidance_scale=1.0 - self._joint_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._joint_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py index 8e15f50f17f6..27f38ca23252 100644 --- a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py @@ -157,7 +157,7 @@ def __init__( force_zeros_for_empty_prompt: bool = True, ): super().__init__() - self._guidance_scale=1.0 + self._guidance_scale = 1.0 # get correct sigmas from LMS scheduler = LMSDiscreteScheduler.from_config(scheduler.config) diff --git a/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py b/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py index 6ec23b72c821..a206c46c3965 100644 --- a/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +++ b/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py @@ -236,11 +236,11 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py b/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py index 9f001f46106b..8935ed05d3d6 100644 --- a/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +++ b/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py @@ -212,11 +212,11 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._num_timesteps = 0 + self._interrupt = False if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py index 6f8157026352..c22ff3d6bd38 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py @@ -256,12 +256,12 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py index e5878c51afc7..57d70227424d 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py @@ -277,13 +277,13 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._denoising_start=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._denoising_start = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py index 6c598eb7d3c6..a26beaa8a241 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py @@ -307,13 +307,13 @@ def __init__( add_watermarker: Optional[bool] = None, ): super().__init__() - self._guidance_scale=1.0 - self._guidance_rescale=0.0 - self._cross_attention_kwargs=None - self._denoising_end=None - self._denoising_start=None - self._num_timesteps=0 - self._interrupt=False + self._guidance_scale = 1.0 + self._guidance_rescale = 0.0 + self._cross_attention_kwargs = None + self._denoising_end = None + self._denoising_start = None + self._num_timesteps = 0 + self._interrupt = False self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py index 139d2f130635..1b44449d2492 100644 --- a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +++ b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py @@ -177,8 +177,8 @@ def __init__( feature_extractor: CLIPImageProcessor, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( vae=vae, 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 8d6e8a3363ac..994ac9f80c89 100644 --- a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +++ b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py @@ -240,7 +240,7 @@ def __init__( requires_safety_checker: bool = True, ): super().__init__() - self._guidance_scale=1.0 + self._guidance_scale = 1.0 if safety_checker is None and requires_safety_checker: logger.warning( diff --git a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py index 68a51f2d8d81..419498b355d0 100644 --- a/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +++ b/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py @@ -288,7 +288,7 @@ def __init__( image_encoder: CLIPVisionModelWithProjection = None, ): super().__init__() - self._guidance_scale=1.0 + self._guidance_scale = 1.0 self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/wan/pipeline_wan.py b/src/diffusers/pipelines/wan/pipeline_wan.py index 1a11ce2e32cb..1b6c2c36e578 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan.py +++ b/src/diffusers/pipelines/wan/pipeline_wan.py @@ -126,11 +126,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._interrupt=False - self._attention_kwargs=None + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._interrupt = False + self._attention_kwargs = None self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/wan/pipeline_wan_i2v.py b/src/diffusers/pipelines/wan/pipeline_wan_i2v.py index b7c3f4eadcf8..41e133069511 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_i2v.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_i2v.py @@ -165,11 +165,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._interrupt=False - self._attention_kwargs=None + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._interrupt = False + self._attention_kwargs = None self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/wan/pipeline_wan_video2video.py b/src/diffusers/pipelines/wan/pipeline_wan_video2video.py index e888ff13e549..821ee2326216 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_video2video.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_video2video.py @@ -205,11 +205,11 @@ def __init__( scheduler: FlowMatchEulerDiscreteScheduler, ): super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 - self._current_timestep=None - self._interrupt=False - self._attention_kwargs=None + self._guidance_scale = 1.0 + self._num_timesteps = 0 + self._current_timestep = None + self._interrupt = False + self._attention_kwargs = None self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py index 99e3c073282b..ef2e354ce279 100644 --- a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py +++ b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py @@ -98,8 +98,8 @@ def __init__( latent_dim_scale: float = 10.67, ) -> None: super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder, diff --git a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py index 4cacb32472ec..958c58181d8d 100644 --- a/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +++ b/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py @@ -116,8 +116,8 @@ def __init__( resolution_multiple: float = 42.67, ) -> None: super().__init__() - self._guidance_scale=1.0 - self._num_timesteps=0 + self._guidance_scale = 1.0 + self._num_timesteps = 0 self.register_modules( tokenizer=tokenizer, text_encoder=text_encoder,