nodes/mcmonkeyprojects-sd-dynamic-thresholding
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

automatically unset DynThresh based on generation parameter sendover

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-02-10 22:17:33 -08:00
parent d0751640ed
commit baedb9e37d
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/dynamic_thresholding.py
+6 -1
View File
@@ -14,7 +14,7 @@ import gradio as gr
import random import random
import torch import torch
import math 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 ######################### ######################### Data values #########################
VALID_MODES = ["Constant", "Linear Down", "Cosine Down", "Half Cosine Down", "Linear Up", "Cosine Up", "Half Cosine Up", "Power Up"] 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_mode, "CFG mode"),
(cfg_scale_min, "CFG scale minimum"), (cfg_scale_min, "CFG scale minimum"),
(power_val, "Power scheduler value")) (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] return [enabled, mimic_scale, threshold_percentile, mimic_mode, mimic_scale_min, cfg_mode, cfg_scale_min, power_val]
last_id = 0 last_id = 0