Skip to content

Commit 603b64b

Browse files
committed
register to config pag
1 parent d5b403f commit 603b64b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/guiders/perturbed_attention_guidance.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
from typing import List, Optional, Union
1616

17+
from ..configuration_utils import register_to_config
1718
from ..hooks import LayerSkipConfig
1819
from .skip_layer_guidance import SkipLayerGuidance
1920

@@ -70,6 +71,7 @@ class PerturbedAttentionGuidance(SkipLayerGuidance):
7071
# complex to support joint latent conditioning in a model-agnostic manner without specializing the implementation
7172
# for each model architecture.
7273

74+
@register_to_config
7375
def __init__(
7476
self,
7577
guidance_scale: float = 7.5,

0 commit comments

Comments
 (0)