Refactor device handling in generate3d function to use CPU instead of CUDA for mesh processing, ensuring compatibility with non-GPU environments. This change improves flexibility in data handling and maintains the integrity of the mesh generation process.

inference.py

@@ -57,8 +57,8 @@ def generate3d(model, rgb, ccm, device):
 
     from kiui.mesh_utils import clean_mesh
     verts, faces = clean_mesh(
-        data_config['verts'].squeeze().cuda().numpy().astype(np.float32),
-        data_config['faces'].squeeze().cuda().numpy().astype(np.int32),
+        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
     )
     data_config['verts'] = torch.from_numpy(verts).to(device).contiguous()