Merge pull request #12522 from catboxanon/fix/extra_params

Restore `extra_params` that was lost in merge
This commit is contained in:
AUTOMATIC1111 2023-08-13 16:38:27 +03:00 committed by GitHub
commit 551d2fabcc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)