Fix WONDER3D error #79

2024-03-07 10:55:44 +01:00
parent 58e500bbef
commit 95dd0a8e20
1 changed files with 1 additions and 1 deletions
@@ -1411,7 +1411,7 @@ class Large_Multiview_Gaussian_Model:
    
    @torch.no_grad()
    def run_LGM(self, multiview_images, lgm_model):
-        device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        device = "cuda"
        ref_image_torch = prepare_torch_img(multiview_images, lgm_model.opt.input_size, lgm_model.opt.input_size, device) # [4, 3, 256, 256]
        ref_image_torch = TF.normalize(ref_image_torch, IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD)
        rays_embeddings = lgm_model.prepare_default_rays(device)