From f7f1208407769fd4844286a09a047bf32af09e23 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Wed, 19 Jul 2023 12:55:17 +0900 Subject: [PATCH] fix: sampler_name --- scripts/!adetailer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 3c0c0f6..90736b4 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -301,11 +301,11 @@ class AfterDetailerScript(scripts.Script): return p.cfg_scale def get_sampler(self, p, args: ADetailerArgs) -> str: - sampler = args.ad_sampler if args.ad_use_sampler else p.sampler + sampler_name = args.ad_sampler if args.ad_use_sampler else p.sampler_name - if sampler in ["PLMS", "UniPC"]: - sampler = "Euler" - return sampler + if sampler_name in ["PLMS", "UniPC"]: + sampler_name = "Euler" + return sampler_name def get_initial_noise_multiplier(self, p, args: ADetailerArgs) -> float | None: if args.ad_use_noise_multiplier: