From a69b71a37f1fd32a60fbd87beed13f4f280400bd Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Wed, 24 May 2023 20:40:37 +0800
Subject: use Schedule instead of Sched

---
 modules/generation_parameters_copypaste.py | 24 ++++++++++++------------
 modules/sd_samplers_kdiffusion.py          |  8 ++++----
 2 files changed, 16 insertions(+), 16 deletions(-)

(limited to 'modules')

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 4f827a6f..1443c5cd 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -306,17 +306,17 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
     if "RNG" not in res:
         res["RNG"] = "GPU"
 
-    if "KDiff Sched Type" not in res:
-        res["KDiff Sched Type"] = "Automatic"
+    if "KDiff Schedule Type" not in res:
+        res["KDiff Schedule Type"] = "Automatic"
 
-    if "KDiff Sched max sigma" not in res:
-        res["KDiff Sched max sigma"] = 14.6
+    if "KDiff Schedule max sigma" not in res:
+        res["KDiff Schedule max sigma"] = 14.6
 
-    if "KDiff Sched min sigma" not in res:
-        res["KDiff Sched min sigma"] = 0.3
+    if "KDiff Schedule min sigma" not in res:
+        res["KDiff Schedule min sigma"] = 0.3
 
-    if "KDiff Sched rho" not in res:
-        res["KDiff Sched rho"] = 7.0
+    if "KDiff Schedule rho" not in res:
+        res["KDiff Schedule rho"] = 7.0
 
     return res
 
@@ -330,10 +330,10 @@ infotext_to_setting_name_mapping = [
     ('Conditional mask weight', 'inpainting_mask_weight'),
     ('Model hash', 'sd_model_checkpoint'),
     ('ENSD', 'eta_noise_seed_delta'),
-    ('KDiff Sched Type', 'k_sched_type'),
-    ('KDiff Sched max sigma', 'sigma_max'),
-    ('KDiff Sched min sigma', 'sigma_min'),
-    ('KDiff Sched rho', 'rho'),
+    ('KDiff Schedule Type', 'k_sched_type'),
+    ('KDiff Schedule max sigma', 'sigma_max'),
+    ('KDiff Schedule min sigma', 'sigma_min'),
+    ('KDiff Schedule rho', 'rho'),
     ('Noise multiplier', 'initial_noise_multiplier'),
     ('Eta', 'eta_ancestral'),
     ('Eta DDIM', 'eta_ddim'),
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index d2d172e4..9c9d9f17 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -328,18 +328,18 @@ class KDiffusionSampler:
             }
 
             sigmas_func = k_diffusion_scheduler[opts.k_sched_type]
-            p.extra_generation_params["KDiff Sched Type"] = opts.k_sched_type
+            p.extra_generation_params["KDiff Schedule Type"] = opts.k_sched_type
 
             if opts.sigma_min != 0.3:
                 # take 0.0 as model default
                 sigmas_kwargs['sigma_min'] = opts.sigma_min or m_sigma_min
-                p.extra_generation_params["KDiff Sched min sigma"] = opts.sigma_min
+                p.extra_generation_params["KDiff Schedule min sigma"] = opts.sigma_min
             if opts.sigma_max != 14.6:
                 sigmas_kwargs['sigma_max'] = opts.sigma_max or m_sigma_max
-                p.extra_generation_params["KDiff Sched max sigma"] = opts.sigma_max
+                p.extra_generation_params["KDiff Schedule max sigma"] = opts.sigma_max
             if opts.k_sched_type != 'exponential':
                 sigmas_kwargs['rho'] = opts.rho
-                p.extra_generation_params["KDiff Sched rho"] = opts.rho
+                p.extra_generation_params["KDiff Schedule rho"] = opts.rho
 
             sigmas = sigmas_func(n=steps, **sigmas_kwargs, device=shared.device)
         elif self.config is not None and self.config.options.get('scheduler', None) == 'karras':
-- 
cgit v1.2.1