From 5fc50239002d1611d2bf5890aa2fbf3af91702d7 Mon Sep 17 00:00:00 2001 From: Ken Simpson Date: Tue, 6 Aug 2024 09:40:54 -0700 Subject: [PATCH] Need to move output tensor to intermediate device before returning from execute() --- image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image.py b/image.py index 5934f29..726a8d0 100644 --- a/image.py +++ b/image.py @@ -1322,7 +1322,7 @@ class ImageColorMatchAdobe(ImageColorMatch): result = fade_factor * transformed + (1 - fade_factor) * image # Convert back to (B, H, W, C) format and ensure values are in [0, 1] range - result = result.permute(0, 2, 3, 1).clamp(0, 1) + result = result.permute(0, 2, 3, 1).clamp(0, 1).to(comfy.model_management.intermediate_device()) return (result,)