1580158
/
sd-dynamic-thresholding
mirror of https://github.com/mcmonkeyprojects/sd-dynamic-thresholding.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add dynthresh params to image metadata 

for #4
pull/9/head
Alex "mcmonkey" Goodwin 2023-02-02 09:18:54 -08:00
parent dfc71c013c
commit 15c81353f1
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/dynamic_thresholding.py
View File

@ -64,6 +64,15 @@ class Script(scripts.Script):
cfg_mode = p.dynthres_cfg_mode if hasattr(p, 'dynthres_cfg_mode') else cfg_mode cfg_mode = p.dynthres_cfg_mode if hasattr(p, 'dynthres_cfg_mode') else cfg_mode
cfg_scale_min = p.dynthres_cfg_scale_min if hasattr(p, 'dynthres_cfg_scale_min') else cfg_scale_min cfg_scale_min = p.dynthres_cfg_scale_min if hasattr(p, 'dynthres_cfg_scale_min') else cfg_scale_min
experiment_mode = p.dynthres_experiment_mode if hasattr(p, 'dynthres_experiment_mode') else 0 experiment_mode = p.dynthres_experiment_mode if hasattr(p, 'dynthres_experiment_mode') else 0
p.extra_generation_params["Dynamic thresholding enabled"] = True
p.extra_generation_params["Mimic scale"] = mimic_scale
p.extra_generation_params["Threshold percentile"] = threshold_percentile
if mimic_mode != "Constant":
p.extra_generation_params["Mimic mode"] = mimic_mode
p.extra_generation_params["Mimic scale minimum"] = mimic_scale_min
if cfg_mode != "Constant":
p.extra_generation_params["CFG mode"] = cfg_mode
p.extra_generation_params["CFG scale minimum"] = cfg_scale_min
# Note: the ID number is to protect the edge case of multiple simultaneous runs with different settings # Note: the ID number is to protect the edge case of multiple simultaneous runs with different settings
Script.last_id += 1 Script.last_id += 1
fixed_sampler_name = f"{p.sampler_name}_dynthres{Script.last_id}" fixed_sampler_name = f"{p.sampler_name}_dynthres{Script.last_id}"