automatically unset DynThresh based on generation parameter sendover

This commit is contained in:
Alex "mcmonkey" Goodwin
2023-02-10 22:17:33 -08:00
parent d0751640ed
commit baedb9e37d
+6 -1
View File
@@ -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