diff --git a/src/transformers/models/electra/configuration_electra.py b/src/transformers/models/electra/configuration_electra.py index 3ea54aa7ca9..01a6d1165a2 100644 --- a/src/transformers/models/electra/configuration_electra.py +++ b/src/transformers/models/electra/configuration_electra.py @@ -119,12 +119,12 @@ class ElectraConfig(PretrainedConfig): Examples: ```python - >>> from transformers import ElectraModel, ElectraConfig + >>> from transformers import ElectraConfig, ElectraModel >>> # Initializing a ELECTRA electra-base-uncased style configuration >>> configuration = ElectraConfig() - >>> # Initializing a model from the electra-base-uncased style configuration + >>> # Initializing a model (with random weights) from the electra-base-uncased style configuration >>> model = ElectraModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 4b7275df8d0..ded7172213f 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -60,6 +60,7 @@ src/transformers/models/detr/configuration_detr.py src/transformers/models/detr/modeling_detr.py src/transformers/models/distilbert/configuration_distilbert.py src/transformers/models/dpt/modeling_dpt.py +src/transformers/models/electra/configuration_electra.py src/transformers/models/electra/modeling_electra.py src/transformers/models/electra/modeling_tf_electra.py src/transformers/models/ernie/configuration_ernie.py