298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-13 11:53:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix orientation of hunyuan 3d model.

Browse Source
This commit is contained in:
comfyanonymous 2025-03-19 19:55:24 -04:00
parent 11f1b41bab
commit 32ca0805b7
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/hunyuan3d/vae.py
View File

@ -581,7 +581,7 @@ class ShapeVAE(nn.Module):
enable_pbar = kwargs.get("enable_pbar", True)
grid_logits = self.volume_decoder(latents, self.geo_decoder, bounds=bounds, num_chunks=num_chunks, octree_resolution=octree_resolution, enable_pbar=enable_pbar)
return grid_logits
return grid_logits.movedim(-2, -1)
def encode(self, x):
return None

5
comfy_extras/nodes_hunyuan3d.py
View File

@ -185,8 +185,8 @@ def voxel_to_mesh(voxels, threshold=0.5, device=None):
device=device
).reshape(-1, 4)
all_indices.append(torch.stack([face_indices[:, 0], face_indices[:, 2], face_indices[:, 1]], dim=1))
all_indices.append(torch.stack([face_indices[:, 0], face_indices[:, 3], face_indices[:, 2]], dim=1))
all_indices.append(torch.stack([face_indices[:, 0], face_indices[:, 1], face_indices[:, 2]], dim=1))
all_indices.append(torch.stack([face_indices[:, 0], face_indices[:, 2], face_indices[:, 3]], dim=1))
vertex_count += 4 * num_faces
@ -202,6 +202,7 @@ def voxel_to_mesh(voxels, threshold=0.5, device=None):
if scale > 0:
vertices = vertices / scale
vertices = torch.fliplr(vertices)
return vertices, faces