Return images as float32

2024-03-26 09:51:37 +02:00
parent e26ddc65ed
commit b1baaf59c4
1 changed files with 2 additions and 2 deletions
@@ -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')