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:
parent
5daf7983d1
commit
45be87afc6
|
@ -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
|
||||||
|
|
|
@ -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)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue