make the program read Eta and Eta DDIM from generation parameters

This commit is contained in:
AUTOMATIC
2023-01-30 10:47:09 +03:00
parent 4df63d2d19
commit 040ec7a80e
4 changed files with 9 additions and 5 deletions

View File

@@ -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):