update

Author: ishan-modi

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):

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)):

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)):

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):

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)
 

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)):

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)):

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

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):

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):