diff --git a/src/transformers/models/ernie/configuration_ernie.py b/src/transformers/models/ernie/configuration_ernie.py index 26a6ab535f5..7fd1c861da7 100644 --- a/src/transformers/models/ernie/configuration_ernie.py +++ b/src/transformers/models/ernie/configuration_ernie.py @@ -96,12 +96,12 @@ class ErnieConfig(PretrainedConfig): Examples: ```python - >>> from transformers import ErnieModel, ErnieConfig + >>> from transformers import ErnieConfig, ErnieModel >>> # Initializing a ERNIE nghuyong/ernie-3.0-base-zh style configuration >>> configuration = ErnieConfig() - >>> # Initializing a model from the nghuyong/ernie-3.0-base-zh style configuration + >>> # Initializing a model (with random weights) from the nghuyong/ernie-3.0-base-zh style configuration >>> model = ErnieModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 531274f3950..76aec4a03a9 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -50,6 +50,7 @@ src/transformers/models/detr/modeling_detr.py src/transformers/models/dpt/modeling_dpt.py src/transformers/models/electra/modeling_electra.py src/transformers/models/electra/modeling_tf_electra.py +src/transformers/models/ernie/configuration_ernie.py src/transformers/models/glpn/modeling_glpn.py src/transformers/models/gpt2/configuration_gpt2.py src/transformers/models/gpt2/modeling_gpt2.py