Fix orientation of hunyuan 3d model.

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

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