update

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_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

@@ -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,

src/diffusers/guiders/tangential_classifier_free_guidance.py

@@ -1911,7 +1911,7 @@ def update(self, **kwargs):
             loader.update(unet=new_unet_model, text_encoder=new_text_encoder)
 
             # Update configuration values
-            loader.update(requires_safety_checker=False, guidance_rescale=0.7)
+            loader.update(requires_safety_checker=False)
 
             # Update both components and configs together
             loader.update(unet=new_unet_model, requires_safety_checker=False)