Merge cc5d177b54 into c6238047ee

Ensure threshold tensor is on the same device as denoise mask.

README.md

@@ -183,7 +183,7 @@ Simply download, extract with [7-Zip](https://7-zip.org) or with the windows exp
 
 If you have trouble extracting it, right click the file -> properties -> unblock
 
-Update your Nvidia drivers if it doesn't start.
+The portable above currently comes with python 3.13 and pytorch cuda 13.0. Update your Nvidia drivers if it doesn't start.
 
 #### Alternative Downloads:
 
@@ -212,7 +212,7 @@ Python 3.14 works but you may encounter issues with the torch compile node. The
 
 Python 3.13 is very well supported. If you have trouble with some custom node dependencies on 3.13 you can try 3.12
 
-torch 2.4 and above is supported but some features might only work on newer versions. We generally recommend using the latest major version of pytorch unless it is less than 2 weeks old.
+torch 2.4 and above is supported but some features might only work on newer versions. We generally recommend using the latest major version of pytorch with the latest cuda version unless it is less than 2 weeks old.
 
 ### Instructions:
 

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)