Fix docs and docstrings Omdet-Turbo (#33726)

Fix weights path in docs
huggingface · Sep 26, 2024 · 77b47e6 · 77b47e6
1 parent c716fc0
commit 77b47e6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/docs/source/en/model_doc/omdet-turbo.md b/docs/source/en/model_doc/omdet-turbo.md
@@ -49,8 +49,8 @@ from PIL import Image
 
 from transformers import AutoProcessor, OmDetTurboForObjectDetection
 
-processor = AutoProcessor.from_pretrained("omlab/omdet-turbo-tiny")
-model = OmDetTurboForObjectDetection.from_pretrained("omlab/omdet-turbo-tiny")
+processor = AutoProcessor.from_pretrained("omlab/omdet-turbo-swin-tiny-hf")
+model = OmDetTurboForObjectDetection.from_pretrained("omlab/omdet-turbo-swin-tiny-hf")
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

diff --git a/src/transformers/models/omdet_turbo/configuration_omdet_turbo.py b/src/transformers/models/omdet_turbo/configuration_omdet_turbo.py
@@ -134,10 +134,10 @@ class OmDetTurboConfig(PretrainedConfig):
     ```python
     >>> from transformers import OmDetTurboConfig, OmDetTurboForObjectDetection
 
-    >>> # Initializing a OmDet-Turbo omlab/omdet-turbo-tiny style configuration
+    >>> # Initializing a OmDet-Turbo omlab/omdet-turbo-swin-tiny-hf style configuration
     >>> configuration = OmDetTurboConfig()
 
-    >>> # Initializing a model (with random weights) from the omlab/omdet-turbo-tiny style configuration
+    >>> # Initializing a model (with random weights) from the omlab/omdet-turbo-swin-tiny-hf style configuration
     >>> model = OmDetTurboForObjectDetection(configuration)
 
     >>> # Accessing the model configuration

diff --git a/src/transformers/models/omdet_turbo/modeling_omdet_turbo.py b/src/transformers/models/omdet_turbo/modeling_omdet_turbo.py
@@ -1709,8 +1709,8 @@ def forward(
 
         >>> from transformers import AutoProcessor, OmDetTurboForObjectDetection
 
-        >>> processor = AutoProcessor.from_pretrained("omlab/omdet-turbo-tiny")
-        >>> model = OmDetTurboForObjectDetection.from_pretrained("omlab/omdet-turbo-tiny")
+        >>> processor = AutoProcessor.from_pretrained("omlab/omdet-turbo-swin-tiny-hf")
+        >>> model = OmDetTurboForObjectDetection.from_pretrained("omlab/omdet-turbo-swin-tiny-hf")
 
         >>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
         >>> image = Image.open(requests.get(url, stream=True).raw)