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

Update dynthres_comfyui.py (#87) 

* 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>
pull/99/head
kaibioinfo 2023-12-22 15:35:27 +01:00 committed by GitHub
parent 249b8cb604
commit bba97c079d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dynthres_comfyui.py
View File

@ -34,8 +34,8 @@ class DynamicThresholdingComfyNode:
cond = input - args["cond"]
uncond = input - args["uncond"]
cond_scale = args["cond_scale"]
time_step = args["timestep"]
dynamic_thresh.step = 999 - time_step[0]
time_step = model.model.model_sampling.timestep(args["sigma"])
dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)
@ -68,8 +68,8 @@ class DynamicThresholdingSimpleComfyNode:
cond = input - args["cond"]
uncond = input - args["uncond"]
cond_scale = args["cond_scale"]
time_step = args["timestep"]
dynamic_thresh.step = 999 - time_step[0]
time_step = model.model.model_sampling.timestep(args["sigma"])
dynamic_thresh.step = 999 - time_step
return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)