Remove kaolin

requirements.txt

@@ -1,5 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cu121
---find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu121.html
 
 torch==2.4.0
 torchvision==0.19.0
@@ -19,7 +18,6 @@ pymeshfix==0.17.0
 igraph==0.11.8
 git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
 xformers==0.0.27.post2
-kaolin==0.17.0
 spconv-cu120==2.3.6
 transformers==4.46.3
 gradio_litmodel3d==0.0.1

trellis/renderers/mesh_renderer.py

@@ -6,11 +6,7 @@
 # distribution of this software and related documentation without an express
 # license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
 import torch
-try:
-    import kaolin as kal
-    import nvdiffrast.torch as dr
-except :
-    print("Kaolin and nvdiffrast are not installed. Please install them to use the mesh renderer.")
+import nvdiffrast.torch as dr
 from easydict import EasyDict as edict
 from ..representations.mesh import MeshExtractResult
 import torch.nn.functional as F

trellis/representations/mesh/cube2mesh.py

@@ -9,10 +9,7 @@ import torch
 from ...modules.sparse import SparseTensor
 from easydict import EasyDict as edict
 from .utils_cube import *
-try:
-    from .flexicube import FlexiCubes
-except:
-    print("Please install kaolin and diso to use the mesh extractor.")
+from .flexicube import FlexiCubes
 
 
 class MeshExtractResult:

trellis/representations/mesh/flexicube.py

@@ -8,7 +8,6 @@
 
 import torch
 from .tables import *
-from kaolin.utils.testing import check_tensor
 
 __all__ = [
     'FlexiCubes'
@@ -49,30 +48,6 @@ class FlexiCubes:
 
     def __call__(self, voxelgrid_vertices, scalar_field, cube_idx, resolution, qef_reg_scale=1e-3,
                  weight_scale=0.99, beta=None, alpha=None, gamma_f=None, voxelgrid_colors=None, training=False):
-        assert torch.is_tensor(voxelgrid_vertices) and \
-            check_tensor(voxelgrid_vertices, (None, 3), throw=False), \
-            "'voxelgrid_vertices' should be a tensor of shape (num_vertices, 3)"
-        num_vertices = voxelgrid_vertices.shape[0]
-        assert torch.is_tensor(scalar_field) and \
-            check_tensor(scalar_field, (num_vertices,), throw=False), \
-            "'scalar_field' should be a tensor of shape (num_vertices,)"
-        assert torch.is_tensor(cube_idx) and \
-            check_tensor(cube_idx, (None, 8), throw=False), \
-            "'cube_idx' should be a tensor of shape (num_cubes, 8)"
-        num_cubes = cube_idx.shape[0]
-        assert beta is None or (
-            torch.is_tensor(beta) and
-            check_tensor(beta, (num_cubes, 12), throw=False)
-        ), "'beta' should be a tensor of shape (num_cubes, 12)"
-        assert alpha is None or (
-            torch.is_tensor(alpha) and
-            check_tensor(alpha, (num_cubes, 8), throw=False)
-        ), "'alpha' should be a tensor of shape (num_cubes, 8)"
-        assert gamma_f is None or (
-            torch.is_tensor(gamma_f) and
-            check_tensor(gamma_f, (num_cubes,), throw=False)
-        ), "'gamma_f' should be a tensor of shape (num_cubes,)"
-
         surf_cubes, occ_fx8 = self._identify_surf_cubes(scalar_field, cube_idx)
         if surf_cubes.sum() == 0:
             return (