From 43c385c26327d1d29429750430720776aa1fcbee Mon Sep 17 00:00:00 2001 From: Niels Date: Tue, 6 Aug 2024 23:00:26 +0200 Subject: [PATCH] Update docstrings --- sam2/sam2_image_predictor.py | 2 +- sam2/sam2_video_predictor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/sam2/sam2_image_predictor.py b/sam2/sam2_image_predictor.py index 6fde883..f6f9a5a 100644 --- a/sam2/sam2_image_predictor.py +++ b/sam2/sam2_image_predictor.py @@ -65,7 +65,7 @@ class SAM2ImagePredictor: @classmethod def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2ImagePredictor": """ - Load a pretrained model from the Hugging Face model hub. + Load a pretrained model from the Hugging Face hub. Arguments: model_id (str): The Hugging Face repository ID. diff --git a/sam2/sam2_video_predictor.py b/sam2/sam2_video_predictor.py index 6e27efb..fe8702b 100644 --- a/sam2/sam2_video_predictor.py +++ b/sam2/sam2_video_predictor.py @@ -106,7 +106,7 @@ class SAM2VideoPredictor(SAM2Base): @classmethod def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2VideoPredictor": """ - Load a pretrained model from the Hugging Face model hub. + Load a pretrained model from the Hugging Face hub. Arguments: model_id (str): The Hugging Face repository ID.