diff --git a/comfy_extras/nodes_differential_diffusion.py b/comfy_extras/nodes_differential_diffusion.py
index 6dfdf466c..d8596077a 100644
--- 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)