Skip to content

Commit 551d2fa

Browse files
Merge pull request #12522 from catboxanon/fix/extra_params
Restore `extra_params` that was lost in merge
2 parents db40d26 + 525b55b commit 551d2fa

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

modules/sd_samplers_kdiffusion.py

+2
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,8 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
6767
def __init__(self, funcname, sd_model, options=None):
6868
super().__init__(funcname)
6969

70+
self.extra_params = sampler_extra_params.get(funcname, [])
71+
7072
self.options = options or {}
7173
self.func = funcname if callable(funcname) else getattr(k_diffusion.sampling, self.funcname)
7274

0 commit comments

Comments
 (0)