yslan commited on
Commit
d73455a
·
1 Parent(s): dcec8da
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import spaces
2
  import os
3
  import torch
 
1
+ import open3d_zerogpu_fix # fix it
2
  import spaces
3
  import os
4
  import torch
nsr/lsgm/flow_matching_trainer.py CHANGED
@@ -126,8 +126,8 @@ from utils.general_utils import matrix_to_quaternion
126
  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 as o3d
131
 
132
  # from sgm.sampling_utils.demo.streamlit_helpers import init_sampling
133
 
 
126
  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 as o3d
131
 
132
  # from sgm.sampling_utils.demo.streamlit_helpers import init_sampling
133
 
nsr/train_nv_util.py CHANGED
@@ -28,9 +28,9 @@ import torch
28
  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 open3d_zerogpu_fix as o3d
34
  import torch.distributed as dist
35
  import torchvision
36
  from PIL import Image
 
28
  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 open3d_zerogpu_fix as o3d
34
  import torch.distributed as dist
35
  import torchvision
36
  from PIL import Image