From d87dcefcf77a3233a669f91daba9e2bc640aea1b Mon Sep 17 00:00:00 2001
From: spacepxl <spacetug93@gmail.com>
Date: Mon, 7 Oct 2024 00:46:13 -0400
Subject: [PATCH] move depth to cpu

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

diff --git a/nodes.py b/nodes.py
index 2d47dcf..a7e074a 100644
--- a/nodes.py
+++ b/nodes.py
@@ -93,7 +93,7 @@ class DepthPro:
         focal_list = focal_px
         focal_avg = np.mean(focal_px)
         
-        return (depth, focal_list, focal_avg)
+        return (depth.to("cpu"), focal_list, focal_avg)
 
 
 class MetricDepthToRelative: