From 4b88e24ebe776680b327e33fe96d7fcf38e2e5d2 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Wed, 24 May 2023 20:35:58 +0800
Subject: improvements

See:
https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/10649#issuecomment-1561047723
---
 modules/generation_parameters_copypaste.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

(limited to 'modules/generation_parameters_copypaste.py')

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index e98866fc..4f827a6f 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -306,6 +306,18 @@ 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 Sched max sigma" not in res:
+        res["KDiff Sched max sigma"] = 14.6
+
+    if "KDiff Sched min sigma" not in res:
+        res["KDiff Sched min sigma"] = 0.3
+
+    if "KDiff Sched rho" not in res:
+        res["KDiff Sched rho"] = 7.0
+
     return res
 
 
@@ -318,10 +330,10 @@ infotext_to_setting_name_mapping = [
     ('Conditional mask weight', 'inpainting_mask_weight'),
     ('Model hash', 'sd_model_checkpoint'),
     ('ENSD', 'eta_noise_seed_delta'),
-    ('KDiffusion Scheduler Type', 'k_sched_type'),
-    ('KDiffusion Scheduler sigma_max', 'sigma_max'),
-    ('KDiffusion Scheduler sigma_min', 'sigma_min'),
-    ('KDiffusion Scheduler rho', 'rho'),
+    ('KDiff Sched Type', 'k_sched_type'),
+    ('KDiff Sched max sigma', 'sigma_max'),
+    ('KDiff Sched min sigma', 'sigma_min'),
+    ('KDiff Sched rho', 'rho'),
     ('Noise multiplier', 'initial_noise_multiplier'),
     ('Eta', 'eta_ancestral'),
     ('Eta DDIM', 'eta_ddim'),
-- 
cgit v1.2.3