update

Author: a-r-r-o-w

src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py

@@ -238,7 +238,7 @@ def __call__(
             components.guider.cleanup_models(components.unet)
 
         # Perform guidance
-        block_state.noise_pred, block_state.scheduler_step_kwargs = components.guider(guider_state)
+        block_state.noise_pred, block_state.guider_inputs = components.guider(guider_state)
 
         return components, block_state
 
@@ -433,7 +433,7 @@ def __call__(self, components: StableDiffusionXLModularPipeline, block_state: Bl
             components.guider.cleanup_models(components.unet)
 
         # Perform guidance
-        block_state.noise_pred, block_state.scheduler_step_kwargs = components.guider(guider_state)
+        block_state.noise_pred, block_state.guider_inputs = components.guider(guider_state)
 
         return components, block_state
 
@@ -492,7 +492,7 @@ def __call__(self, components: StableDiffusionXLModularPipeline, block_state: Bl
             t,
             block_state.latents,
             **block_state.extra_step_kwargs,
-            **block_state.scheduler_step_kwargs,
+            **block_state.guider_inputs,
             return_dict=False,
         )[0]
 
@@ -590,7 +590,7 @@ def __call__(self, components: StableDiffusionXLModularPipeline, block_state: Bl
             t,
             block_state.latents,
             **block_state.extra_step_kwargs,
-            **block_state.scheduler_step_kwargs,
+            **block_state.guider_inputs,
             return_dict=False,
         )[0]
 

src/diffusers/modular_pipelines/wan/denoise.py

@@ -127,7 +127,7 @@ def __call__(
             components.guider.cleanup_models(components.transformer)
 
         # Perform guidance
-        block_state.noise_pred, block_state.scheduler_step_kwargs = components.guider(guider_state)
+        block_state.noise_pred, block_state.guider_inputs = components.guider(guider_state)
 
         return components, block_state
 
@@ -171,7 +171,7 @@ def __call__(self, components: WanModularPipeline, block_state: BlockState, i: i
             block_state.noise_pred.float(),
             t,
             block_state.latents.float(),
-            **block_state.scheduler_step_kwargs,
+            **block_state.guider_inputs,
             return_dict=False,
         )[0]