From 841cc1f0154adf4fd10e0bf0fbbdede62c6698e7 Mon Sep 17 00:00:00 2001 From: Niels Date: Mon, 5 Aug 2024 09:44:03 +0200 Subject: [PATCH] Update docstring --- sam2/sam2_video_predictor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sam2/sam2_video_predictor.py b/sam2/sam2_video_predictor.py index d687bc1..3a751c5 100644 --- a/sam2/sam2_video_predictor.py +++ b/sam2/sam2_video_predictor.py @@ -104,7 +104,7 @@ class SAM2VideoPredictor(SAM2Base): 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 @@ class SAM2VideoPredictor(SAM2Base): **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