Explorar el Código

Update docstrings

Niels hace 1 año
padre
commit
43c385c263
Se han modificado 2 ficheros con 2 adiciones y 2 borrados
  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.