Add Trellis 2 Support (CORE-7) (#12183)

This commit is contained in:
Yousef R. Gamaleldin
2026-05-21 17:01:38 +03:00
committed by GitHub
parent 4259a0c7c3
commit 7547897b6f
17 changed files with 4891 additions and 18 deletions

View File

@@ -7,9 +7,10 @@ import torch
class VOXEL:
def __init__(self, data: torch.Tensor):
def __init__(self, data: torch.Tensor, voxel_colors=None, resolution=None):
self.data = data
self.voxel_colors = voxel_colors
self.resolution = resolution # each 3d model has its own resolution
class MESH:
def __init__(self, vertices: torch.Tensor, faces: torch.Tensor,