diff --git a/docs/source/en/model_doc/omdet-turbo.md b/docs/source/en/model_doc/omdet-turbo.md index 190ac3e31eeaae..1e9e05a898d230 100644 --- 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 index cb5e69db5f901a..8b4a620a17acc4 100644 --- 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 index bb6c8838ff8c65..b40f09c4e8f2a6 100644 --- 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)