Merge remote-tracking branch 'KohakuBlueleaf/custom-k-sched-settings' into dev

1 files changed, 16 insertions, 0 deletions

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index d5f0a49b..1443c5cd 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 Schedule Type" not in res:

+        res["KDiff Schedule Type"] = "Automatic"

+

+    if "KDiff Schedule max sigma" not in res:

+        res["KDiff Schedule max sigma"] = 14.6

+

+    if "KDiff Schedule min sigma" not in res:

+        res["KDiff Schedule min sigma"] = 0.3

+

+    if "KDiff Schedule rho" not in res:

+        res["KDiff Schedule rho"] = 7.0

+

     return res

 

 

@@ -318,6 +330,10 @@ infotext_to_setting_name_mapping = [
     ('Conditional mask weight', 'inpainting_mask_weight'),

     ('Model hash', 'sd_model_checkpoint'),

     ('ENSD', 'eta_noise_seed_delta'),

+    ('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'),