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

skip dynthresh if mimic == cfg 

per chaotic reddit post https://www.reddit.com/r/StableDiffusion/comments/1enxcek/comment
pull/108/head
Alex "mcmonkey" Goodwin 2024-08-10 05:59:56 -07:00
parent 1a49c55c1a
commit 5d63447afb
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dynthres_comfyui.py
View File

@ -38,7 +38,10 @@ class DynamicThresholdingComfyNode:
time_step = time_step[0].item() time_step = time_step[0].item()
dynamic_thresh.step = 999 - time_step dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None) if cond_scale == mimic_scale:
return input - (uncond + (cond - uncond) * cond_scale)
else:
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)
m = model.clone() m = model.clone()
m.set_model_sampler_cfg_function(sampler_dyn_thresh) m.set_model_sampler_cfg_function(sampler_dyn_thresh)
@ -73,7 +76,10 @@ class DynamicThresholdingSimpleComfyNode:
time_step = time_step[0].item() time_step = time_step[0].item()
dynamic_thresh.step = 999 - time_step dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None) if cond_scale == mimic_scale:
return input - (uncond + (cond - uncond) * cond_scale)
else:
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)
m = model.clone() m = model.clone()
m.set_model_sampler_cfg_function(sampler_dyn_thresh) m.set_model_sampler_cfg_function(sampler_dyn_thresh)