xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-10 05:15:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ensure colormatch is on cpu

Browse Source
This commit is contained in:
Kijai 2023-12-20 15:47:45 +02:00
parent a77d73ca8a
commit 1448402b83
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -1152,8 +1152,8 @@ class ColorMatch:
images_target = image_target.squeeze() images_target = image_target.squeeze()
images_ref = image_ref.squeeze() images_ref = image_ref.squeeze()
image_ref_np = images_ref.numpy() image_ref_np = images_ref.cpu().numpy()
images_target_np = images_target.numpy() images_target_np = images_target.cpu().numpy()
if image_ref.size(0) > 1 and image_ref.size(0) != batch_size: 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.") raise ValueError("ColorMatch: Use either single reference image or a matching batch of reference images.")