Spaces:
mashroo
/
Runtime error

YoussefAnso commited on
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
Files changed (1) hide show
  1. 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).cuda().contiguous()
157
- data_config['faces'] = torch.from_numpy(faces).cuda().contiguous()
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():