Ver Fonte

Update docstrings

Niels há 1 ano atrás
pai
commit
43c385c263
2 ficheiros alterados com 2 adições e 2 exclusões
  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.