update

Author: a-r-r-o-w

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/guider_utils.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__"