Fix some issues with --gpu-only

2025-09-11 03:58:22 +00:00 · 2024-01-27 02:51:27 -05:00
parent fc196aac80
commit 7f4725f6b3
2 changed files with 2 additions and 1 deletions
--- a/comfy_extras/nodes_post_processing.py
+++ b/comfy_extras/nodes_post_processing.py
@@ -33,6 +33,7 @@ class Blend:
    CATEGORY = "image/postprocessing"

    def blend_images(self, image1: torch.Tensor, image2: torch.Tensor, blend_factor: float, blend_mode: str):
+        image2 = image2.to(image1.device)
        if image1.shape != image2.shape:
            image2 = image2.permute(0, 3, 1, 2)
            image2 = comfy.utils.common_upscale(image2, image1.shape[2], image1.shape[1], upscale_method='bicubic', crop='center')