diff options
author | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-08-13 13:08:34 +0000 |
---|---|---|
committer | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-08-13 13:08:34 +0000 |
commit | 525b55b1e9dc589a1133a8031ed3b1646e5cccff (patch) | |
tree | b4f9698c89dcc84555eaf39201bfb5763c190b76 /modules/sd_samplers_kdiffusion.py | |
parent | d8419762c1454ba51baa710d9ce8e762efc056ef (diff) | |
download | stable-diffusion-webui-gfx803-525b55b1e9dc589a1133a8031ed3b1646e5cccff.tar.gz stable-diffusion-webui-gfx803-525b55b1e9dc589a1133a8031ed3b1646e5cccff.tar.bz2 stable-diffusion-webui-gfx803-525b55b1e9dc589a1133a8031ed3b1646e5cccff.zip |
Restore extra_params that was lost in merge
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
-rw-r--r-- | modules/sd_samplers_kdiffusion.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index 1f8e9c4b..9a89e7fa 100644 --- 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)
|