Merge pull request #12522 from catboxanon/fix/extra_params

Restore `extra_params` that was lost in merge
AUTOMATIC1111 · Aug 13, 2023 · 551d2fa · 551d2fa
2 parents db40d26 + 525b55b
commit 551d2fa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
@@ -67,6 +67,8 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
     def __init__(self, funcname, sd_model, options=None):
         super().__init__(funcname)
 
+        self.extra_params = sampler_extra_params.get(funcname, [])
+
         self.options = options or {}
         self.func = funcname if callable(funcname) else getattr(k_diffusion.sampling, self.funcname)