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,