Modular Guider ConfigMixin (#11862)

* update

* update

* register to config pag

Author: a-r-r-o-w

src/diffusers/guiders/adaptive_projected_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from .guider_utils import BaseGuidance, rescale_noise_cfg
 
 
@@ -53,6 +54,7 @@ class AdaptiveProjectedGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/auto_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from ..hooks import HookRegistry, LayerSkipConfig
 from ..hooks.layer_skip import _apply_layer_skip_hook
 from .guider_utils import BaseGuidance, rescale_noise_cfg
@@ -60,6 +61,7 @@ class AutoGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/classifier_free_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from .guider_utils import BaseGuidance, rescale_noise_cfg
 
 
@@ -67,6 +68,7 @@ class ClassifierFreeGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/classifier_free_zero_star_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from .guider_utils import BaseGuidance, rescale_noise_cfg
 
 
@@ -58,6 +59,7 @@ class ClassifierFreeZeroStarGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/entropy_rectifying_guidance.py

@@ -16,19 +16,24 @@
 
 import torch
 
+from ..configuration_utils import ConfigMixin
 from ..utils import get_logger
 
 
 if TYPE_CHECKING:
     from ..modular_pipelines.modular_pipeline import BlockState
 
 
+GUIDER_CONFIG_NAME = "guider_config.json"
+
+
 logger = get_logger(__name__)  # pylint: disable=invalid-name
 
 
-class BaseGuidance:
+class BaseGuidance(ConfigMixin):
     r"""Base class providing the skeleton for implementing guidance techniques."""
 
+    config_name = GUIDER_CONFIG_NAME
     _input_predictions = None
     _identifier_key = "__guidance_identifier__"
 

src/diffusers/guiders/guider_utils.py

@@ -14,6 +14,7 @@
 
 from typing import List, Optional, Union
 
+from ..configuration_utils import register_to_config
 from ..hooks import LayerSkipConfig
 from .skip_layer_guidance import SkipLayerGuidance
 
@@ -70,6 +71,7 @@ class PerturbedAttentionGuidance(SkipLayerGuidance):
     # complex to support joint latent conditioning in a model-agnostic manner without specializing the implementation
     # for each model architecture.
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/perturbed_attention_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from ..hooks import HookRegistry, LayerSkipConfig
 from ..hooks.layer_skip import _apply_layer_skip_hook
 from .guider_utils import BaseGuidance, rescale_noise_cfg
@@ -86,6 +87,7 @@ class SkipLayerGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond", "pred_cond_skip"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/skip_layer_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from ..hooks import HookRegistry
 from ..hooks.smoothed_energy_guidance_utils import SmoothedEnergyGuidanceConfig, _apply_smoothed_energy_guidance_hook
 from .guider_utils import BaseGuidance, rescale_noise_cfg
@@ -76,6 +77,7 @@ class SmoothedEnergyGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond", "pred_cond_seg"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,

src/diffusers/guiders/smoothed_energy_guidance.py

@@ -17,6 +17,7 @@
 
 import torch
 
+from ..configuration_utils import register_to_config
 from .guider_utils import BaseGuidance, rescale_noise_cfg
 
 
@@ -49,6 +50,7 @@ class TangentialClassifierFreeGuidance(BaseGuidance):
 
     _input_predictions = ["pred_cond", "pred_uncond"]
 
+    @register_to_config
     def __init__(
         self,
         guidance_scale: float = 7.5,