From 020c017e02bf0b2ba6708102752fdc22dcc75301 Mon Sep 17 00:00:00 2001 From: unknown Date: Sun, 16 Nov 2025 22:52:03 +0100 Subject: [PATCH] Fix output types --- color_transfer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/color_transfer.py b/color_transfer.py index a3afc26..e125bf8 100644 --- a/color_transfer.py +++ b/color_transfer.py @@ -438,7 +438,7 @@ class PaletteOptimalTransportTransfer(ComfyNodeABC): recolored_image = recolored_pixels.reshape(h, w, 3) - result_tensor = torch.from_numpy(recolored_image).unsqueeze(0) + result_tensor = torch.from_numpy(recolored_image).float().unsqueeze(0) processed_images.append(result_tensor) return (torch.cat(processed_images, dim=0),)