瀏覽代碼

Update docstrings

Niels 1 年之前
父節點
當前提交
43c385c263
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1 1
      sam2/sam2_image_predictor.py
  2. 1 1
      sam2/sam2_video_predictor.py

+ 1 - 1
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.

+ 1 - 1
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.