From baedb9e37df1c4a3eee392fa526fee787f4a7079 Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" Date: Fri, 10 Feb 2023 22:17:33 -0800 Subject: [PATCH] automatically unset DynThresh based on generation parameter sendover --- scripts/dynamic_thresholding.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/dynamic_thresholding.py b/scripts/dynamic_thresholding.py index ada0279..69d330a 100644 --- a/scripts/dynamic_thresholding.py +++ b/scripts/dynamic_thresholding.py @@ -14,7 +14,7 @@ import gradio as gr import random import torch import math -from modules import scripts, sd_samplers, sd_samplers_kdiffusion, sd_samplers_common +from modules import scripts, sd_samplers, sd_samplers_kdiffusion, sd_samplers_common, script_callbacks ######################### Data values ######################### VALID_MODES = ["Constant", "Linear Down", "Cosine Down", "Half Cosine Down", "Linear Up", "Cosine Up", "Half Cosine Up", "Power Up"] @@ -56,6 +56,11 @@ class Script(scripts.Script): (cfg_mode, "CFG mode"), (cfg_scale_min, "CFG scale minimum"), (power_val, "Power scheduler value")) + # Add implicit DynThresh=False + def onInfotext(infotext: str, result: dict[str, any]): + if "Dynamic thresholding enabled" not in result: + result["Dynamic thresholding enabled"] = False + script_callbacks.on_infotext_pasted(onInfotext) return [enabled, mimic_scale, threshold_percentile, mimic_mode, mimic_scale_min, cfg_mode, cfg_scale_min, power_val] last_id = 0