Commit
·
82982c9
1
Parent(s):
9551df9
Change device setting for mesh data in generate3d function to CPU for consistency with recent updates, ensuring compatibility across systems without GPU support.
Browse files- inference.py +2 -2
inference.py
CHANGED
@@ -153,8 +153,8 @@ def generate3d(model, rgb, ccm, device):
|
|
153 |
|
154 |
from kiui.mesh_utils import clean_mesh
|
155 |
verts, faces = clean_mesh(data_config['verts'].squeeze().cpu().numpy().astype(np.float32), data_config['faces'].squeeze().cpu().numpy().astype(np.int32), repair = False, remesh=True, remesh_size=0.005, remesh_iters=1)
|
156 |
-
data_config['verts'] = torch.from_numpy(verts).
|
157 |
-
data_config['faces'] = torch.from_numpy(faces).
|
158 |
|
159 |
start_time = time.time()
|
160 |
with torch.no_grad():
|
|
|
153 |
|
154 |
from kiui.mesh_utils import clean_mesh
|
155 |
verts, faces = clean_mesh(data_config['verts'].squeeze().cpu().numpy().astype(np.float32), data_config['faces'].squeeze().cpu().numpy().astype(np.int32), repair = False, remesh=True, remesh_size=0.005, remesh_iters=1)
|
156 |
+
data_config['verts'] = torch.from_numpy(verts).to('cpu').contiguous()
|
157 |
+
data_config['faces'] = torch.from_numpy(faces).to('cpu').contiguous()
|
158 |
|
159 |
start_time = time.time()
|
160 |
with torch.no_grad():
|