auto-restore mimic mode / cfg mode when loading from metadata

for #34
2023-05-08 16:05:13 -07:00
parent 5d3465c4b2
commit 019aa9beab
1 changed files with 2 additions and 2 deletions
@@ -62,8 +62,8 @@ class Script(scripts.Script):
            (mimic_scale, "Mimic scale"),
            (threshold_percentile, "Threshold percentile"),
            (mimic_scale_min, "Mimic scale minimum"),
-            (mimic_mode, "Mimic mode"),
-            (cfg_mode, "CFG mode"),
+            (mimic_mode, lambda d: gr.Dropdown.update(value=d.get("Mimic mode", "Constant"))),
+            (cfg_mode, lambda d: gr.Dropdown.update(value=d.get("CFG mode", "Constant"))),
            (cfg_scale_min, "CFG scale minimum"),
            (power_val, "Power scheduler value"))
        return [enabled, mimic_scale, threshold_percentile, mimic_mode, mimic_scale_min, cfg_mode, cfg_scale_min, power_val]