PUM4CH3N commited on
Commit
42e27b9
·
1 Parent(s): 08ac8df

disable o3d

Browse files
trellis/pipelines/trellis_text_to_3d.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  import torch.nn as nn
4
  import numpy as np
5
  from transformers import CLIPTextModel, AutoTokenizer
6
- import open3d as o3d
7
  from .base import Pipeline
8
  from . import samplers
9
  from ..modules import sparse as sp
@@ -230,7 +230,7 @@ class TrellisTextTo3DPipeline(Pipeline):
230
  slat = self.sample_slat(cond, coords, slat_sampler_params)
231
  print(4)
232
  return self.decode_slat(slat, formats)
233
-
234
  def voxelize(self, mesh: o3d.geometry.TriangleMesh) -> torch.Tensor:
235
  """
236
  Voxelize a mesh.
@@ -281,3 +281,4 @@ class TrellisTextTo3DPipeline(Pipeline):
281
  torch.manual_seed(seed)
282
  slat = self.sample_slat(cond, coords, slat_sampler_params)
283
  return self.decode_slat(slat, formats)
 
 
3
  import torch.nn as nn
4
  import numpy as np
5
  from transformers import CLIPTextModel, AutoTokenizer
6
+ # import open3d as o3d
7
  from .base import Pipeline
8
  from . import samplers
9
  from ..modules import sparse as sp
 
230
  slat = self.sample_slat(cond, coords, slat_sampler_params)
231
  print(4)
232
  return self.decode_slat(slat, formats)
233
+ '''
234
  def voxelize(self, mesh: o3d.geometry.TriangleMesh) -> torch.Tensor:
235
  """
236
  Voxelize a mesh.
 
281
  torch.manual_seed(seed)
282
  slat = self.sample_slat(cond, coords, slat_sampler_params)
283
  return self.decode_slat(slat, formats)
284
+ '''