From 95dd0a8e206e8ecc7c9148f5d79f11e195441e79 Mon Sep 17 00:00:00 2001
From: MrForExample <62230687+MrForExample@users.noreply.github.com>
Date: Thu, 7 Mar 2024 10:55:44 +0100
Subject: [PATCH] Fix WONDER3D error #79

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)