diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-01-30 07:47:09 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-30 07:47:09 +0000 |
commit | 040ec7a80e23d340efe1108b9de5ead62d9011a9 (patch) | |
tree | abb66ecdc936a08f7b83efc1653d55ae052b9709 /modules/sd_samplers_compvis.py | |
parent | 4df63d2d197f26181758b5108f003f225fe84874 (diff) | |
download | stable-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.py | 3 |
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):
|