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

Update dynthres_comfyui.py (#87)

Browse Source 
* Update dynthres_comfyui.py

comfyui timestep arg returns the sigma instead of the timestep. Thanks to piezo for helping me figure out how to convert sigma to timestep.

* fix over-deletion

---------

Co-authored-by: Alex "mcmonkey" Goodwin <git_commits@alexgoodwin.dev>
This commit is contained in:
kaibioinfo
2023-12-22 06:35:27 -08:00
committed by GitHub
co-authored by Alex mcmonkey Goodwin
parent 249b8cb604
commit bba97c079d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dynthres_comfyui.py
+4 -4
View File
@@ -34,8 +34,8 @@ class DynamicThresholdingComfyNode:
cond = input - args["cond"] cond = input - args["cond"]
uncond = input - args["uncond"] uncond = input - args["uncond"]
cond_scale = args["cond_scale"] cond_scale = args["cond_scale"]
time_step = args["timestep"] time_step = model.model.model_sampling.timestep(args["sigma"])
dynamic_thresh.step = 999 - time_step[0] dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None) return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)
@@ -68,8 +68,8 @@ class DynamicThresholdingSimpleComfyNode:
cond = input - args["cond"] cond = input - args["cond"]
uncond = input - args["uncond"] uncond = input - args["uncond"]
cond_scale = args["cond_scale"] cond_scale = args["cond_scale"]
time_step = args["timestep"] time_step = model.model.model_sampling.timestep(args["sigma"])
dynamic_thresh.step = 999 - time_step[0] dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None) return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)