rename enable/disable

Author: a-r-r-o-w

src/diffusers/guiders/guider_utils.py

@@ -57,10 +57,10 @@ def __init__(self, start: float = 0.0, stop: float = 1.0):
                 "`_input_predictions` must be a list of required prediction names for the guidance technique."
             )
 
-    def _force_disable(self):
+    def disable(self):
         self._enabled = False
 
-    def _force_enable(self):
+    def enable(self):
         self._enabled = True
 
     def set_state(self, step: int, num_inference_steps: int, timestep: torch.LongTensor) -> None:

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_modular.py

@@ -2231,9 +2231,9 @@ def __call__(self, pipeline, state: PipelineState) -> PipelineState:
         data.num_channels_unet = pipeline.unet.config.in_channels
         data.disable_guidance = True if pipeline.unet.config.time_cond_proj_dim is not None else False
         if data.disable_guidance:
-            pipeline.guider._force_disable()
+            pipeline.guider.disable()
         else:
-            pipeline.guider._force_enable()
+            pipeline.guider.enable()
 
         # Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
         data.extra_step_kwargs = self.prepare_extra_step_kwargs(pipeline, data.generator, data.eta)
@@ -2611,9 +2611,9 @@ def __call__(self, pipeline, state: PipelineState) -> PipelineState:
         # (2) Prepare conditional inputs for unet using the guider
         data.disable_guidance = True if pipeline.unet.config.time_cond_proj_dim is not None else False
         if data.disable_guidance:
-            pipeline.guider._force_disable()
+            pipeline.guider.disable()
         else:
-            pipeline.guider._force_enable()
+            pipeline.guider.enable()
 
         # (4) Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
         data.extra_step_kwargs = self.prepare_extra_step_kwargs(pipeline, data.generator, data.eta)
@@ -3011,9 +3011,9 @@ def __call__(self, pipeline, state: PipelineState) -> PipelineState:
         # adding default guider arguments: disable_guidance, guidance_scale, guidance_rescale
         data.disable_guidance = True if pipeline.unet.config.time_cond_proj_dim is not None else False
         if data.disable_guidance:
-            pipeline.guider._force_disable()
+            pipeline.guider.disable()
         else:
-            pipeline.guider._force_enable()
+            pipeline.guider.enable()
 
         data.control_type = data.control_type.reshape(1, -1).to(data.device, dtype=data.prompt_embeds.dtype)
         repeat_by = data.batch_size * data.num_images_per_prompt // data.control_type.shape[0]