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

Move threshold tensor to denoise mask device

Browse Source 
Ensure threshold tensor is on the same device as denoise mask.
This commit is contained in:
jzhang-POP 2026-01-06 16:41:58 +01:00
parent 3cd7b32f1b
commit cc5d177b54
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)