From 7dc9d9e27e157ccd7a0c59405a3aaf4f1ca7e194 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Mon, 22 May 2023 23:34:16 +0800
Subject: [PATCH] only add metadata when k_sched is actually been used

---
 modules/processing.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/modules/processing.py b/modules/processing.py
index 260a573a..5e50f1d6 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -13,7 +13,7 @@ from skimage import exposure
 from typing import Any, Dict, List
 
 import modules.sd_hijack
-from modules import devices, prompt_parser, masking, sd_samplers, lowvram, generation_parameters_copypaste, extra_networks, sd_vae_approx, scripts, sd_samplers_common
+from modules import devices, prompt_parser, masking, sd_samplers, lowvram, generation_parameters_copypaste, extra_networks, sd_vae_approx, scripts, sd_samplers_common, sd_samplers_kdiffusion
 from modules.sd_hijack import model_hijack
 from modules.shared import opts, cmd_opts, state
 import modules.shared as shared
@@ -560,14 +560,16 @@ def create_infotext(p, all_prompts, all_seeds, all_subseeds, comments=None, iter
     if uses_ensd:
         uses_ensd = sd_samplers_common.is_sampler_using_eta_noise_seed_delta(p)
 
+    use_custom_k_sched = p.enable_custom_k_sched and p.sampler_name in sd_samplers_kdiffusion.k_diffusion_samplers_map
+
     generation_params = {
         "Steps": p.steps,
         "Sampler": p.sampler_name,
-        "Enable Custom Karras Schedule": p.enable_custom_k_sched or None,
-        "kdiffusion Scheduler Type": p.k_sched_type if p.enable_custom_k_sched else None,
-        "kdiffusion Scheduler sigma_max": p.sigma_max if p.enable_custom_k_sched else None,
-        "kdiffusion Scheduler sigma_min": p.sigma_min if p.enable_custom_k_sched else None,
-        "kdiffusion Scheduler rho": p.rho if p.enable_custom_k_sched else None,
+        "Enable Custom Karras Schedule": use_custom_k_sched or None,
+        "kdiffusion Scheduler Type": p.k_sched_type if use_custom_k_sched else None,
+        "kdiffusion Scheduler sigma_max": p.sigma_max if use_custom_k_sched else None,
+        "kdiffusion Scheduler sigma_min": p.sigma_min if use_custom_k_sched else None,
+        "kdiffusion Scheduler rho": p.rho if use_custom_k_sched else None,
         "CFG scale": p.cfg_scale,
         "Image CFG scale": getattr(p, 'image_cfg_scale', None),
         "Seed": all_seeds[index],