update

asset3d_gen/data/backproject_v2.py

@@ -425,7 +425,8 @@ class TextureBacker:
         mesh: trimesh.Trimesh,
     ) -> trimesh.Trimesh:
         self._lazy_init_render(self.camera_params, self.mask_thresh)
-
+        mesh = self.load_mesh(mesh)
+        print("self.device cuda_forward", self.device)
         vertices = torch.from_numpy(mesh.vertices).to(self.device).float()
         faces = torch.from_numpy(mesh.faces).to(self.device).to(torch.int)
         uv_map = torch.from_numpy(mesh.visual.uv).to(self.device).float()
@@ -463,7 +464,7 @@ class TextureBacker:
         mesh: trimesh.Trimesh,
         output_path: str,
     ) -> trimesh.Trimesh:
-        mesh = self.load_mesh(mesh)
+        
 
         texture_np, mask_np = self.cuda_forward(colors, mesh)