Skip to content

Commit b9fd002

Browse files
committed
add distilled_guidance_scale to adp
1 parent 1e045af commit b9fd002

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/diffusers/guiders/adaptive_projected_guidance_mix.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@ def __init__(
7575
stop: float = 1.0,
7676
adaptive_projected_guidance_start_step: int = 5,
7777
enabled: bool = True,
78+
distilled_guidance_scale: Optional[float] = None,
7879
):
7980
super().__init__(start, stop, enabled)
8081

@@ -87,6 +88,7 @@ def __init__(
8788
self.adaptive_projected_guidance_start_step = adaptive_projected_guidance_start_step
8889
self.use_original_formulation = use_original_formulation
8990
self.momentum_buffer = None
91+
self.distilled_guidance_scale = distilled_guidance_scale
9092

9193
def prepare_inputs(
9294
self, data: "BlockState", input_fields: Optional[Dict[str, Union[str, Tuple[str, str]]]] = None

0 commit comments

Comments
 (0)