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: