Commit 45be87af authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

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

parent 5daf7983

modules/sd_samplers_common.py

+2 −1
Original line number Diff line number Diff line
@@ -217,6 +217,7 @@ class Sampler: 


        self.eta_option_field = 'eta_ancestral'

        self.eta_infotext_field = 'Eta'

        self.eta_default = 1.0



        self.conditioning_key = shared.sd_model.model.conditioning_key
@@ -273,7 +274,7 @@ class Sampler: 
                extra_params_kwargs[param_name] = getattr(p, param_name)



        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



            extra_params_kwargs['eta'] = self.eta

modules/sd_samplers_timesteps.py

+1 −0
Original line number Diff line number Diff line
@@ -76,6 +76,7 @@ class CompVisSampler(sd_samplers_common.Sampler): 


        self.eta_option_field = 'eta_ddim'

        self.eta_infotext_field = 'Eta DDIM'

        self.eta_default = 0.0



        self.model_wrap_cfg = CFGDenoiserTimesteps(self)

CRA Git | Maintained and supported by SUSTech CRA and CCSE