aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_compvis.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-30 07:47:09 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-01-30 07:47:09 +0000
commit040ec7a80e23d340efe1108b9de5ead62d9011a9 (patch)
treeabb66ecdc936a08f7b83efc1653d55ae052b9709 /modules/sd_samplers_compvis.py
parent4df63d2d197f26181758b5108f003f225fe84874 (diff)
downloadstable-diffusion-webui-gfx803-040ec7a80e23d340efe1108b9de5ead62d9011a9.tar.gz
stable-diffusion-webui-gfx803-040ec7a80e23d340efe1108b9de5ead62d9011a9.tar.bz2
stable-diffusion-webui-gfx803-040ec7a80e23d340efe1108b9de5ead62d9011a9.zip
make the program read Eta and Eta DDIM from generation parameters
Diffstat (limited to 'modules/sd_samplers_compvis.py')
-rw-r--r--modules/sd_samplers_compvis.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_samplers_compvis.py b/modules/sd_samplers_compvis.py
index 88541193..d03131cd 100644
--- a/modules/sd_samplers_compvis.py
+++ b/modules/sd_samplers_compvis.py
@@ -27,7 +27,6 @@ class VanillaStableDiffusionSampler:
self.step = 0
self.stop_at = None
self.eta = None
- self.default_eta = 0.0
self.config = None
self.last_latent = None
@@ -102,6 +101,8 @@ class VanillaStableDiffusionSampler:
def initialize(self, p):
self.eta = p.eta if p.eta is not None else shared.opts.eta_ddim
+ if self.eta != 0.0:
+ p.extra_generation_params["Eta DDIM"] = self.eta
for fieldname in ['p_sample_ddim', 'p_sample_plms']:
if hasattr(self.sampler, fieldname):