SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix docs and docstrings Omdet-Turbo (#33726)

Browse Source 
Fix weights path in docs
This commit is contained in:
Yoni Gozlan
2024-09-26 12:18:23 -04:00
committed by GitHub
parent c716fc0e48
commit 77b47e6645
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/en/model_doc/omdet-turbo.md
View File

@@ -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)

4
src/transformers/models/omdet_turbo/configuration_omdet_turbo.py
View File

@@ -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

4
src/transformers/models/omdet_turbo/modeling_omdet_turbo.py
View File

@@ -1709,8 +1709,8 @@ class OmDetTurboForObjectDetection(OmDetTurboPreTrainedModel):
>>> 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)