aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_kdiffusion.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-13 13:38:27 +0000
committerGitHub <noreply@github.com>2023-08-13 13:38:27 +0000
commit551d2fabcc6b8b550821192a91b62280de9a705a (patch)
tree54503e2c9ea1efa6386945af7dd1f3339b6109f3 /modules/sd_samplers_kdiffusion.py
parentdb40d26d086b6e980bd28ce7eb3008ec09b1f0e8 (diff)
parent525b55b1e9dc589a1133a8031ed3b1646e5cccff (diff)
downloadstable-diffusion-webui-gfx803-551d2fabcc6b8b550821192a91b62280de9a705a.tar.gz
stable-diffusion-webui-gfx803-551d2fabcc6b8b550821192a91b62280de9a705a.tar.bz2
stable-diffusion-webui-gfx803-551d2fabcc6b8b550821192a91b62280de9a705a.zip
Merge pull request #12522 from catboxanon/fix/extra_params
Restore `extra_params` that was lost in merge
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
-rw-r--r--modules/sd_samplers_kdiffusion.py2
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)