Update DAM_V2/depth_anything_v2/dpt.py

DAM_V2/depth_anything_v2/dpt.py

@@ -215,9 +215,9 @@ class DepthAnythingV2(nn.Module):
         image = transform({'image': image})['image']
         image = torch.from_numpy(image).unsqueeze(0)
         
-        DEVICE = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu'
+        # DEVICE = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu'
 
-        # DEVICE = "cpu"
+        DEVICE = "cpu"
         image = image.to(DEVICE)
         
         return image, (h, w)