wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-28 03:12:31 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

fix: issue 88 texture path resolution and gpu-only host conversion

Browse Source
This commit is contained in:
John Pollock 2026-04-20 14:35:08 -05:00
parent 880d7823e8
commit 06661522d9
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_trellis2.py
View File

@ -111,8 +111,8 @@ def paint_mesh_with_voxels(mesh, voxel_coords, voxel_colors, resolution):
verts = mesh.vertices.to(device).squeeze(0) verts = mesh.vertices.to(device).squeeze(0)
voxel_colors = voxel_colors.to(device) voxel_colors = voxel_colors.to(device)
voxel_pos_np = voxel_pos.numpy() voxel_pos_np = voxel_pos.cpu().numpy()
verts_np = verts.numpy() verts_np = verts.cpu().numpy()
tree = scipy.spatial.cKDTree(voxel_pos_np) tree = scipy.spatial.cKDTree(voxel_pos_np)
@ -194,6 +194,7 @@ class VaeDecodeTextureTrellis(IO.ComfyNode):
IO.Latent.Input("samples"), IO.Latent.Input("samples"),
IO.Vae.Input("vae"), IO.Vae.Input("vae"),
IO.AnyType.Input("shape_subs"), IO.AnyType.Input("shape_subs"),
IO.Combo.Input("resolution", options=["512", "1024"], default="1024")
], ],
outputs=[ outputs=[
IO.Mesh.Output("mesh"), IO.Mesh.Output("mesh"),
@ -201,9 +202,9 @@ class VaeDecodeTextureTrellis(IO.ComfyNode):
) )
@classmethod @classmethod
def execute(cls, shape_mesh, samples, vae, shape_subs): def execute(cls, shape_mesh, samples, vae, shape_subs, resolution):
resolution = 1024 resolution = int(resolution)
patcher = vae.patcher patcher = vae.patcher
device = comfy.model_management.get_torch_device() device = comfy.model_management.get_torch_device()
comfy.model_management.load_model_gpu(patcher) comfy.model_management.load_model_gpu(patcher)