From 91d83f519fd4efd4abb7cf0cad19ff9a100ce9b2 Mon Sep 17 00:00:00 2001
From: Kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 25 Mar 2024 18:36:22 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index acd77c9..03178a1 100644
--- a/nodes.py
+++ b/nodes.py
@@ -174,9 +174,10 @@ class geowizard_sampler:
 
             depth_out = torch.cat(depth_maps, dim=0)
             depth_out = torch.clamp(depth, 0.0, 1.0)
-            depth_out = 1.0 - depth
+            depth_out = 1.0 - depth_out
+            depth_out = depth_out.cpu()
 
-            normal_out = torch.cat(normal_maps, dim=0)
+            normal_out = torch.cat(normal_maps, dim=0).cpu()
 
             return (depth_out, normal_out)