another sparsity issue fixed

marigold_dc.py

@@ -185,7 +185,7 @@ class MarigoldDepthCompletionPipeline(MarigoldDepthPipeline):
             with torch.no_grad():
                 pred_latent.data = self.scheduler.step(noise, t, pred_latent, generator=generator).prev_sample
 
-            yield current_metric_estimate, rmse.item()
+            yield current_metric_estimate, rmse
 
             del (
                 pred_original_sample,