Ensure tensors are on CPU in generate3d function of inference.py for consistent processing. Updated points_tensor and triplane_feature2 to prevent device-related issues during inference.

inference.py

@@ -81,7 +81,8 @@ def generate3d(model, rgb, ccm, device):
     mesh_trimesh = trimesh.Trimesh(vertices=mesh_v, faces=mesh_f, process=False)
     points, face_indices = trimesh.sample.sample_surface(mesh_trimesh, tex_res[0]*tex_res[1])
     # Use the model's decoder and rgbMlp to get color for each sampled point
-    points_tensor = torch.from_numpy(points).float().unsqueeze(0)  # (1, N, 3)
+    points_tensor = torch.from_numpy(points).float().unsqueeze(0).cpu()  # (1, N, 3) on CPU
+    triplane_feature2 = triplane_feature2.cpu()  # Ensure on CPU
     with torch.no_grad():
         dec_verts = model.decoder(triplane_feature2, points_tensor)
         colors = model.rgbMlp(dec_verts).squeeze().cpu().numpy()  # (N, 3)