update

asset3d_gen/data/backproject_v2.py

@@ -425,8 +425,7 @@ 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()
@@ -464,8 +463,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)
         
         texture_np = self.uv_inpaint(mesh, texture_np, mask_np)