riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats 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