From 61260cad8f14842aad463938a74fb9d30b6b3bb5 Mon Sep 17 00:00:00 2001
From: "Alex \"mcmonkey\" Goodwin" <git_commits@alexgoodwin.dev>
Date: Fri, 10 Feb 2023 06:50:12 -0800
Subject: [PATCH] put power scheduler in params when relevant

---
 scripts/dynamic_thresholding.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/dynamic_thresholding.py b/scripts/dynamic_thresholding.py
index 88e4fb7..928c953 100644
--- a/scripts/dynamic_thresholding.py
+++ b/scripts/dynamic_thresholding.py
@@ -74,6 +74,8 @@ class Script(scripts.Script):
         if cfg_mode != "Constant":
             p.extra_generation_params["CFG mode"] = cfg_mode
             p.extra_generation_params["CFG scale minimum"] = cfg_scale_min
+        if cfg_mode == "Power Up" or mimic_mode == "Power Up":
+            p.extra_generation_params["Power scheduler value"] = power_val
         # Note: the ID number is to protect the edge case of multiple simultaneous runs with different settings
         Script.last_id += 1
         fixed_sampler_name = f"{p.sampler_name}_dynthres{Script.last_id}"