diff --git a/nodes.py b/nodes.py index 87e841b..5ac7397 100644 --- a/nodes.py +++ b/nodes.py @@ -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)