yslan commited on
Commit
dcec8da
·
1 Parent(s): ab78067
nsr/lsgm/flow_matching_trainer.py CHANGED
@@ -127,7 +127,7 @@ from utils.mesh_util import post_process_mesh, to_cam_open3d_compat
127
  from datasets.g_buffer_objaverse import focal2fov, fov2focal
128
 
129
  # import open3d as o3d
130
- import open3d_zerogpu_fix
131
 
132
  # from sgm.sampling_utils.demo.streamlit_helpers import init_sampling
133
 
 
127
  from datasets.g_buffer_objaverse import focal2fov, fov2focal
128
 
129
  # import open3d as o3d
130
+ import open3d_zerogpu_fix as o3d
131
 
132
  # from sgm.sampling_utils.demo.streamlit_helpers import init_sampling
133
 
nsr/train_nv_util.py CHANGED
@@ -29,7 +29,8 @@ from torch.autograd import Function
29
  import torch.nn.functional as F
30
  import torch as th
31
  # import open3d as o3d
32
- import open3d_zerogpu_fix
 
33
  import torch.distributed as dist
34
  import torchvision
35
  from PIL import Image
 
29
  import torch.nn.functional as F
30
  import torch as th
31
  # import open3d as o3d
32
+ # import open3d_zerogpu_fix
33
+ import open3d_zerogpu_fix as o3d
34
  import torch.distributed as dist
35
  import torchvision
36
  from PIL import Image
utils/mesh_util.py CHANGED
@@ -14,7 +14,8 @@ import numpy as np
14
  from PIL import Image
15
  from functools import partial
16
  # import open3d as o3d
17
- import open3d_zerogpu_fix
 
18
  import trimesh
19
 
20
 
 
14
  from PIL import Image
15
  from functools import partial
16
  # import open3d as o3d
17
+ # import open3d_zerogpu_fix
18
+ import open3d_zerogpu_fix as o3d
19
  import trimesh
20
 
21