wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-09 05:40:49 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Merge cc5d177b54 into 3cd7b32f1b

Browse Source
This commit is contained in:
jzhang-POP 2026-01-07 12:47:34 +01:00 committed by GitHub
commit 3550ef1216
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes_differential_diffusion.py
View File

@ -49,6 +49,9 @@ class DifferentialDiffusion(io.ComfyNode):
threshold = (current_ts - ts_to) / (ts_from - ts_to)
# Move the threshold tensor to the same device as denoise mask tensor
threshold = threshold.to(denoise_mask.device)
# Generate the binary mask based on the threshold
binary_mask = (denoise_mask >= threshold).to(denoise_mask.dtype)