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

[Doctest] Add configuration_yolos.py (#19539)

Browse Source 
* YOLOS Config for doctest

* fix
This commit is contained in:
Partho
2022-10-12 23:31:25 +05:30
committed by GitHub
parent 209bec4636
commit 51d21b7619
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/yolos/configuration_yolos.py
View File

@@ -92,12 +92,12 @@ class YolosConfig(PretrainedConfig):
Example:
```python
>>> from transformers import YolosModel, YolosConfig
>>> from transformers import YolosConfig, YolosModel
>>> # Initializing a YOLOS hustvl/yolos-base style configuration
>>> configuration = YolosConfig()
>>> # Initializing a model from the hustvl/yolos-base style configuration
>>> # Initializing a model (with random weights) from the hustvl/yolos-base style configuration
>>> model = YolosModel(configuration)
>>> # Accessing the model configuration

1
utils/documentation_tests.txt
View File

@@ -101,5 +101,6 @@ src/transformers/models/wav2vec2_with_lm/processing_wav2vec2_with_lm.py
src/transformers/models/wavlm/modeling_wavlm.py
src/transformers/models/whisper/modeling_whisper.py
src/transformers/models/whisper/modeling_tf_whisper.py
src/transformers/models/yolos/configuration_yolos.py
src/transformers/models/yolos/modeling_yolos.py
src/transformers/models/x_clip/modeling_x_clip.py