diff --git a/src/diffusers/callbacks.py b/src/diffusers/callbacks.py index 10d8c44d85af..4b8b15368c47 100644 --- a/src/diffusers/callbacks.py +++ b/src/diffusers/callbacks.py @@ -97,8 +97,8 @@ def callback_fn(self, pipeline, step_index, timestep, callback_kwargs) -> Dict[s class SDXLCFGCutoffCallback(PipelineCallback): """ - Callback function for the base Stable Diffusion XL Pipelines. After certain number of steps (set by `cutoff_step_ratio` - or `cutoff_step_index`), this callback will disable the CFG. + Callback function for the base Stable Diffusion XL Pipelines. After certain number of steps (set by + `cutoff_step_ratio` or `cutoff_step_index`), this callback will disable the CFG. Note: This callback mutates the pipeline by changing the `_guidance_scale` attribute to 0.0 after the cutoff step. """ @@ -139,8 +139,8 @@ def callback_fn(self, pipeline, step_index, timestep, callback_kwargs) -> Dict[s class SDXLControlnetCFGCutoffCallback(PipelineCallback): """ - Callback function for the Controlnet Stable Diffusion XL Pipelines. After certain number of steps (set by `cutoff_step_ratio` - or `cutoff_step_index`), this callback will disable the CFG. + Callback function for the Controlnet Stable Diffusion XL Pipelines. After certain number of steps (set by + `cutoff_step_ratio` or `cutoff_step_index`), this callback will disable the CFG. Note: This callback mutates the pipeline by changing the `_guidance_scale` attribute to 0.0 after the cutoff step. """