Merge cc5d177b54 into 1a20656448

Move threshold tensor to denoise mask device
Ensure threshold tensor is on the same device as denoise mask.
2026-01-24 05:10:18 +08:00 · 2026-01-08 16:49:05 -08:00 · 2026-01-07 12:47:32 +01:00
1 changed files with 3 additions and 0 deletions
--- a/comfy_extras/nodes_differential_diffusion.py
+++ b/comfy_extras/nodes_differential_diffusion.py
@ -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)
Author	SHA1	Message	Date
jzhang-POP	06445d4ded	Merge `cc5d177b54` into `1a20656448`	2026-01-08 16:49:05 -08:00
jzhang-POP	cc5d177b54	Move threshold tensor to denoise mask device Ensure threshold tensor is on the same device as denoise mask.	2026-01-07 12:47:32 +01:00