Can't bake texture

I'm trying to generate models with textures.

RTX4080, one graphic card.

2025-03-01 16:56:44,490 - INFO - Baking texture ... Traceback (most recent call last): File "D:\WorkProgram\TripoSR\run.py", line 181, in <module> bake_output = bake_texture(meshes[0], model, scene_codes[0], args.texture_resolution) File "D:\WorkProgram\TripoSR\tsr\bake_texture.py", line 162, in bake_texture colors_texture = positions_to_colors( File "D:\WorkProgram\TripoSR\tsr\bake_texture.py", line 140, in positions_to_colors queried_grid = model.renderer.query_triplane( File "D:\WorkProgram\TripoSR\tsr\models\nerf_renderer.py", line 78, in query_triplane net_out = chunk_batch(_query_chunk, self.chunk_size, positions) File "D:\WorkProgram\TripoSR\tsr\\\utils.py", line 169, in chunk_batch out_chunk = func( File "D:\WorkProgram\TripoSR\tsr\models\nerf_renderer.py", line 61, in _query_chunk out: torch.Tensor = F.grid_sample( File "D:\WorkProgram\TripoSR\lib\site-packages\torch\nn\functional.py", line 5023, in grid_sample return torch.grid_sampler(input, grid, mode_enum, padding_mode_enum, align_corners) RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument grid in method wrapper_CUDA__grid_sampler_2d)