fix: sampler_name

Cryptik-Rick · Jul 19, 2023 · f7f1208 · f7f1208
1 parent e697b54
commit f7f1208
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
@@ -301,11 +301,11 @@ def get_cfg_scale(self, p, args: ADetailerArgs) -> float:
         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: