From b1baaf59c4138fea00a14ca9093e6d102e5530a9 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 26 Mar 2024 09:51:37 +0200
Subject: [PATCH] Return images as float32

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

diff --git a/nodes.py b/nodes.py
index a6740bd..da80c3c 100644
--- a/nodes.py
+++ b/nodes.py
@@ -199,9 +199,9 @@ 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_out
-            depth_out = depth_out.cpu()
+            depth_out = depth_out.cpu().to(torch.float32)
 
-            normal_out = torch.cat(normal_maps, dim=0).cpu()
+            normal_out = torch.cat(normal_maps, dim=0).cpu().to(torch.float32)
             if not keep_model_loaded:
                 pipe = pipe.to('cpu')