correctly add Eta DDIM to infotext when it's 1.0 and do not add it when it's 0.0.

This commit is contained in:
AUTOMATIC1111 2023-08-14 21:48:05 +03:00
parent 5daf7983d1
commit 45be87afc6
2 changed files with 3 additions and 1 deletions

View File

@ -217,6 +217,7 @@ class Sampler:
self.eta_option_field = 'eta_ancestral' self.eta_option_field = 'eta_ancestral'
self.eta_infotext_field = 'Eta' self.eta_infotext_field = 'Eta'
self.eta_default = 1.0
self.conditioning_key = shared.sd_model.model.conditioning_key self.conditioning_key = shared.sd_model.model.conditioning_key
@ -273,7 +274,7 @@ class Sampler:
extra_params_kwargs[param_name] = getattr(p, param_name) extra_params_kwargs[param_name] = getattr(p, param_name)
if 'eta' in inspect.signature(self.func).parameters: if 'eta' in inspect.signature(self.func).parameters:
if self.eta != 1.0: if self.eta != self.eta_default:
p.extra_generation_params[self.eta_infotext_field] = self.eta p.extra_generation_params[self.eta_infotext_field] = self.eta
extra_params_kwargs['eta'] = self.eta extra_params_kwargs['eta'] = self.eta

View File

@ -76,6 +76,7 @@ class CompVisSampler(sd_samplers_common.Sampler):
self.eta_option_field = 'eta_ddim' self.eta_option_field = 'eta_ddim'
self.eta_infotext_field = 'Eta DDIM' self.eta_infotext_field = 'Eta DDIM'
self.eta_default = 0.0
self.model_wrap_cfg = CFGDenoiserTimesteps(self) self.model_wrap_cfg = CFGDenoiserTimesteps(self)