From 7b95f7c4b05414d60e893c40a3f9d2b5b9e58732 Mon Sep 17 00:00:00 2001 From: John Pollock Date: Mon, 20 Apr 2026 14:39:15 -0500 Subject: [PATCH] fix: issue 88 unify texture paint color path on host --- comfy_extras/nodes_trellis2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy_extras/nodes_trellis2.py b/comfy_extras/nodes_trellis2.py index e00444f8c..6651ea72a 100644 --- a/comfy_extras/nodes_trellis2.py +++ b/comfy_extras/nodes_trellis2.py @@ -109,7 +109,7 @@ def paint_mesh_with_voxels(mesh, voxel_coords, voxel_colors, resolution): # map voxels voxel_pos = voxel_coords.to(device).float() * voxel_size + origin verts = mesh.vertices.to(device).squeeze(0) - voxel_colors = voxel_colors.to(device) + voxel_colors = voxel_colors.cpu() voxel_pos_np = voxel_pos.cpu().numpy() verts_np = verts.cpu().numpy()