Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files- nsr/lsgm/flow_matching_trainer.py +1 -1
- nsr/train_nv_util.py +2 -1
- utils/mesh_util.py +2 -1
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 |
|