aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_kdiffusion.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-12 09:39:59 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-12 09:39:59 +0000
commit64311faa6848d641cc452115e4e1eb47d2a7b519 (patch)
tree3849ad4ca8ad7f44c8f20e3ab359578f3a4021ec /modules/sd_samplers_kdiffusion.py
parent26c92f056acc795af5066779f1b8aedb8dfa983d (diff)
downloadstable-diffusion-webui-gfx803-64311faa6848d641cc452115e4e1eb47d2a7b519.tar.gz
stable-diffusion-webui-gfx803-64311faa6848d641cc452115e4e1eb47d2a7b519.tar.bz2
stable-diffusion-webui-gfx803-64311faa6848d641cc452115e4e1eb47d2a7b519.zip
put refiner into main UI, into the new accordions section
add VAE from main model into infotext, not from refiner model option to make scripts UI without gr.Group fix inconsistencies with refiner when usings samplers that do more denoising than steps
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
-rw-r--r--modules/sd_samplers_kdiffusion.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index d10fe12e..1f8e9c4b 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -64,9 +64,10 @@ class CFGDenoiserKDiffusion(sd_samplers_cfg_denoiser.CFGDenoiser):
class KDiffusionSampler(sd_samplers_common.Sampler):
- def __init__(self, funcname, sd_model):
+ def __init__(self, funcname, sd_model, options=None):
super().__init__(funcname)
+ self.options = options or {}
self.func = funcname if callable(funcname) else getattr(k_diffusion.sampling, self.funcname)
self.model_wrap_cfg = CFGDenoiserKDiffusion(self)