fix device bug

app.py

@@ -87,7 +87,7 @@ model = model.half().to(device)
 model.eval()
 
 tan_half_fov = np.tan(0.5 * np.deg2rad(opt.fovy))
-proj_matrix = torch.zeros(4, 4, dtype=torch.float32, device=device)
+proj_matrix = torch.zeros(4, 4, dtype=torch.float32).to(device)
 proj_matrix[0, 0] = 1 / tan_half_fov
 proj_matrix[1, 1] = 1 / tan_half_fov
 proj_matrix[2, 2] = (opt.zfar + opt.znear) / (opt.zfar - opt.znear)

core/models.py

@@ -134,7 +134,7 @@ class LTRFM_NeRF(nn.Module):
 
         if not use_texture_map:
             # query vertex colors
-            vertices_tensor = torch.tensor(vertices, dtype=torch.float32, device=device).unsqueeze(0)
+            vertices_tensor = torch.tensor(vertices, dtype=torch.float32).to(device).unsqueeze(0)
             rgb_colors = self.tensorRF.predict_color(
                 planes, vertices_tensor)['rgb'].squeeze(0).cpu().numpy()
             rgb_colors = (rgb_colors * 255).astype(np.uint8)