xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-05-29 06:46:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-03-19 10:47:30 +02:00
parent 2253869d04
commit b96bb5b6d8
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@ -1158,7 +1158,10 @@ class Hy3DGenerateMeshMultiView():
images.append(rgb) images.append(rgb)
image_tensors = torch.cat(images, 0).permute(0, 2, 3, 1).cpu().float() image_tensors = torch.cat(images, 0).permute(0, 2, 3, 1).cpu().float()
mask_tensors = torch.cat(masks, 0).squeeze(1).cpu().float() if masks:
mask_tensors = torch.cat(masks, 0).squeeze(1).cpu().float()
else:
mask_tensors = torch.zeros(image_tensors.shape[0], image_tensors.shape[1], image_tensors.shape[2]).cpu().float()
pipeline.to(offload_device) pipeline.to(offload_device)