add UniPC options to infotext

1 files changed, 14 insertions, 0 deletions

diff --git a/modules/sd_samplers_compvis.py b/modules/sd_samplers_compvis.py
index 7d07c4a5..083da18c 100644
--- a/modules/sd_samplers_compvis.py
+++ b/modules/sd_samplers_compvis.py
@@ -129,6 +129,19 @@ class VanillaStableDiffusionSampler:
         if self.eta != 0.0:

             p.extra_generation_params["Eta DDIM"] = self.eta

 

+        if self.is_unipc:

+            keys = [

+                ('UniPC variant', 'uni_pc_variant'),

+                ('UniPC skip type', 'uni_pc_skip_type'),

+                ('UniPC order', 'uni_pc_order'),

+                ('UniPC lower order final', 'uni_pc_lower_order_final'),

+            ]

+

+            for name, key in keys:

+                v = getattr(shared.opts, key)

+                if v != shared.opts.get_default(key):

+                    p.extra_generation_params[name] = v

+

         for fieldname in ['p_sample_ddim', 'p_sample_plms']:

             if hasattr(self.sampler, fieldname):

                 setattr(self.sampler, fieldname, self.p_sample_ddim_hook)

@@ -138,6 +151,7 @@ class VanillaStableDiffusionSampler:
         self.mask = p.mask if hasattr(p, 'mask') else None

         self.nmask = p.nmask if hasattr(p, 'nmask') else None

 

+

     def adjust_steps_if_invalid(self, p, num_steps):

         if ((self.config.name == 'DDIM') and p.ddim_discretize == 'uniform') or (self.config.name == 'PLMS') or (self.config.name == 'UniPC'):

             if self.config.name == 'UniPC' and num_steps < shared.opts.uni_pc_order: