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