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

[Doctest] SqueezeBERT Config for doctest (#19758)

Browse Source
This commit is contained in:
Partho
2022-10-20 13:46:39 +05:30
committed by GitHub
parent cc03063366
commit 123f65eea6
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/squeezebert/configuration_squeezebert.py
View File

@@ -94,12 +94,12 @@ class SqueezeBertConfig(PretrainedConfig):
Examples:
```python
>>> from transformers import SqueezeBertModel, SqueezeBertConfig
>>> from transformers import SqueezeBertConfig, SqueezeBertModel
>>> # Initializing a SqueezeBERT configuration
>>> configuration = SqueezeBertConfig()
>>> # Initializing a model from the configuration above
>>> # Initializing a model (with random weights) from the configuration above
>>> model = SqueezeBertModel(configuration)
>>> # Accessing the model configuration

1
utils/documentation_tests.txt
View File

@@ -125,6 +125,7 @@ src/transformers/models/speech_to_text/configuration_speech_to_text.py
src/transformers/models/speech_to_text/modeling_speech_to_text.py
src/transformers/models/speech_to_text_2/modeling_speech_to_text_2.py
src/transformers/models/segformer/modeling_tf_segformer.py
src/transformers/models/squeezebert/configuration_squeezebert.py
src/transformers/models/swin/configuration_swin.py
src/transformers/models/swin/modeling_swin.py
src/transformers/models/swinv2/configuration_swinv2.py