Update docstring

rjrthnm2 · Aug 5, 2024 · 841cc1f · 841cc1f
1 parent e93be7f
commit 841cc1f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sam2/sam2_video_predictor.py b/sam2/sam2_video_predictor.py
@@ -104,7 +104,7 @@ def init_state(
         return inference_state
 
     @classmethod
-    def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2ImagePredictor":
+    def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2VideoPredictor":
         """
         Load a pretrained model from the Hugging Face model hub.
 
@@ -113,7 +113,7 @@ def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2ImagePredictor":
           **kwargs: Additional arguments to pass to the model constructor.
 
         Returns:
-          (SAM2ImagePredictor): The loaded model.
+          (SAM2VideoPredictor): The loaded model.
         """
         from sam2.build_sam import build_sam2_video_predictor_hf