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

fix comfy large batch for #88

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-12-31 09:26:27 -08:00
parent bba97c079d
commit 0d017bcf44
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dynthres_comfyui.py
+2
View File
@@ -35,6 +35,7 @@ class DynamicThresholdingComfyNode:
uncond = input - args["uncond"]
cond_scale = args["cond_scale"]
time_step = model.model.model_sampling.timestep(args["sigma"])
time_step = time_step[0].item()
dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)
@@ -69,6 +70,7 @@ class DynamicThresholdingSimpleComfyNode:
uncond = input - args["uncond"]
cond_scale = args["cond_scale"]
time_step = model.model.model_sampling.timestep(args["sigma"])
time_step = time_step[0].value()
dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)