style

huggingface · Aug 29, 2024 · adeb1cc · adeb1cc
1 parent 7f5ece9
commit adeb1cc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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.
     """