infotext support for #14978

Author: AUTOMATIC1111

modules/infotext_utils.py

@@ -359,6 +359,9 @@ def parse_generation_parameters(x: str, skip_fields: list[str] | None = None):
     if "Emphasis" not in res:
         res["Emphasis"] = "Original"
 
+    if "Refiner switch by sampling steps" not in res:
+        res["Refiner switch by sampling steps"] = False
+
     infotext_versions.backcompat(res)
 
     for key in skip_fields:

modules/infotext_versions.py

@@ -5,6 +5,7 @@
 
 v160 = version.parse("1.6.0")
 v170_tsnr = version.parse("v1.7.0-225")
+v180 = version.parse("1.8.0")
 
 
 def parse_version(text):
@@ -40,3 +41,5 @@ def backcompat(d):
     if ver < v170_tsnr:
         d["Downcast alphas_cumprod"] = True
 
+    if ver < v180 and d.get('Refiner'):
+        d["Refiner switch by sampling steps"] = True

modules/sd_samplers_common.py

@@ -155,14 +155,16 @@ def torchsde_randn(size, dtype, device, seed):
 replace_torchsde_browinan()
 
 
-def apply_refiner(cfg_denoiser, sigma):
-    if opts.refiner_switch_by_sample_steps:
+def apply_refiner(cfg_denoiser, sigma=None):
+    if opts.refiner_switch_by_sample_steps or not sigma:
         completed_ratio = cfg_denoiser.step / cfg_denoiser.total_steps
+        cfg_denoiser.p.extra_generation_params["Refiner switch by sampling steps"] = True
+
     else:
         # torch.max(sigma) only to handle rare case where we might have different sigmas in the same batch
         try:
             timestep = torch.argmin(torch.abs(cfg_denoiser.inner_model.sigmas - torch.max(sigma)))
-        except AttributeError: # for samplers that dont use sigmas (DDIM) sigma is actually the timestep
+        except AttributeError:  # for samplers that don't use sigmas (DDIM) sigma is actually the timestep
             timestep = torch.max(sigma).to(dtype=int)
         completed_ratio = (999 - timestep) / 1000