From a8c1a707a92c98346d68e3a79adcb810d011c9bd Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 21 Dec 2023 02:06:32 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index 26a825a..cddcb6c 100644 --- a/nodes.py +++ b/nodes.py @@ -1147,13 +1147,15 @@ class ColorMatch: def colormatch(self, image_ref, image_target, method): cm = ColorMatcher() + image_ref = image_ref.cpu() + image_target = image_target.cpu() batch_size = image_target.size(0) out = [] images_target = image_target.squeeze() images_ref = image_ref.squeeze() - image_ref_np = images_ref.cpu().numpy() - images_target_np = images_target.cpu().numpy() + image_ref_np = images_ref.numpy() + images_target_np = images_target.numpy() if image_ref.size(0) > 1 and image_ref.size(0) != batch_size: raise ValueError("ColorMatch: Use either single reference image or a matching batch of reference images.")