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

[Doctest] fix doc test for megatron bert (#19600)

Browse Source
This commit is contained in:
RamitPahwa
2022-10-14 05:08:55 -05:00
committed by GitHub
parent c7d1fb6964
commit 7036c956fe
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/megatron_bert/configuration_megatron_bert.py
View File

@@ -77,12 +77,12 @@ class MegatronBertConfig(PretrainedConfig):
Examples: Examples:
```python ```python
>>> from transformers import MegatronBertModel, MegatronBertConfig >>> from transformers import MegatronBertConfig, MegatronBertModel
>>> # Initializing a MEGATRON_BERT bert-base-uncased style configuration >>> # Initializing a MEGATRON_BERT bert-base-uncased style configuration
>>> configuration = MegatronBertConfig() >>> configuration = MegatronBertConfig()
>>> # Initializing a model from the bert-base-uncased style configuration >>> # Initializing a model (with random weights) from the bert-base-uncased style configuration
>>> model = MegatronBertModel(configuration) >>> model = MegatronBertModel(configuration)
>>> # Accessing the model configuration >>> # Accessing the model configuration

1
utils/documentation_tests.txt
View File

@@ -68,6 +68,7 @@ src/transformers/models/longt5/modeling_longt5.py
src/transformers/models/marian/modeling_marian.py src/transformers/models/marian/modeling_marian.py
src/transformers/models/markuplm/modeling_markuplm.py src/transformers/models/markuplm/modeling_markuplm.py
src/transformers/models/mbart/modeling_mbart.py src/transformers/models/mbart/modeling_mbart.py
src/transformers/models/megatron_bert/configuration_megatron_bert.py
src/transformers/models/mobilebert/configuration_mobilebert.py src/transformers/models/mobilebert/configuration_mobilebert.py
src/transformers/models/mobilebert/modeling_mobilebert.py src/transformers/models/mobilebert/modeling_mobilebert.py
src/transformers/models/mobilebert/modeling_tf_mobilebert.py src/transformers/models/mobilebert/modeling_tf_mobilebert.py