diff --git a/segment_anything/predictor.py b/segment_anything/predictor.py
index 91b2ed3..6661d2a 100644
--- a/segment_anything/predictor.py
+++ b/segment_anything/predictor.py
@@ -97,7 +97,7 @@ class SamPredictor:
         mask_input: Optional[np.ndarray] = None,
         multimask_output: bool = True,
         return_logits: bool = False,
-    ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
+    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
         """
         Predict masks for the given input prompts, using the currently set image.