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

smarter sampler ID handling

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-01-27 18:49:19 -08:00
parent 3beb86f32f
commit cf524a9dbc
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/dynamic_thresholding.py
+12 -4
View File
@@ -47,6 +47,8 @@ class Script(scripts.Script):
show_progress = False) show_progress = False)
return [enabled, mimic_scale, threshold_percentile, mimic_mode, cfg_mode] return [enabled, mimic_scale, threshold_percentile, mimic_mode, cfg_mode]
last_id = 0
def process_batch(self, p, enabled, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, batch_number, prompts, seeds, subseeds): def process_batch(self, p, enabled, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, batch_number, prompts, seeds, subseeds):
enabled = p.dynthres_enabled if hasattr(p, 'dynthres_enabled') else enabled enabled = p.dynthres_enabled if hasattr(p, 'dynthres_enabled') else enabled
if not enabled: if not enabled:
@@ -57,9 +59,9 @@ class Script(scripts.Script):
threshold_percentile = p.dynthres_threshold_percentile if hasattr(p, 'dynthres_threshold_percentile') else threshold_percentile threshold_percentile = p.dynthres_threshold_percentile if hasattr(p, 'dynthres_threshold_percentile') else threshold_percentile
mimic_mode = p.dynthres_mimic_mode if hasattr(p, 'dynthres_mimic_mode') else mimic_mode mimic_mode = p.dynthres_mimic_mode if hasattr(p, 'dynthres_mimic_mode') else mimic_mode
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
# Note: the random 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
fixed_sampler_name = f"{p.sampler_name}_dynthres{random.randrange(100)}" Script.last_id += 1
p.fixed_sampler_name = fixed_sampler_name fixed_sampler_name = f"{p.sampler_name}_dynthres{Script.last_id}"
# Percentage to portion # Percentage to portion
threshold_percentile *= 0.01 threshold_percentile *= 0.01
# Make a placeholder sampler # Make a placeholder sampler
@@ -70,13 +72,19 @@ class Script(scripts.Script):
result.model_wrap_cfg = cfg result.model_wrap_cfg = cfg
return result return result
newSampler = sd_samplers.SamplerData(fixed_sampler_name, newConstructor, sampler.aliases, sampler.options) newSampler = sd_samplers.SamplerData(fixed_sampler_name, newConstructor, sampler.aliases, sampler.options)
# Apply for usage
p.orig_sampler_name = p.sampler_name
p.sampler_name = fixed_sampler_name p.sampler_name = fixed_sampler_name
p.fixed_sampler_name = fixed_sampler_name
sd_samplers.all_samplers_map[fixed_sampler_name] = newSampler sd_samplers.all_samplers_map[fixed_sampler_name] = newSampler
def postprocess_batch(self, p, enabled, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, batch_number, images): def postprocess_batch(self, p, enabled, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, batch_number, images):
if not enabled: if not enabled or not hasattr(p, 'orig_sampler_name'):
return return
p.sampler_name = p.orig_sampler_name
del sd_samplers.all_samplers_map[p.fixed_sampler_name] del sd_samplers.all_samplers_map[p.fixed_sampler_name]
del p.orig_sampler_name
del p.fixed_sampler_name
######################### Implementation logic ######################### ######################### Implementation logic #########################