TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

[alboto]

when ComfyUI use param --gpu-only will get this error

08-20 13:17:08.107 Traceback (most recent call last):
08-20 13:17:08.107 File "/app/plugins/comfyui/execution.py", line 81, in map_node_over_list
08-20 13:17:08.107 results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
08-20 13:17:08.107 File "/app/plugins/comfyui/custom_nodes/comfyui-art-venture/modules/postprocessing/color_correct.py", line 64, in color_correct
08-20 13:17:08.107 tensor_image = image[b].numpy()
08-20 13:17:08.107 TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.