From 10ea45b902055b14a6db5f9718d3f4c7134d4003 Mon Sep 17 00:00:00 2001 From: Jing Hua <59118459+ztjhz@users.noreply.github.com> Date: Fri, 14 Oct 2022 22:57:51 +0800 Subject: [PATCH] Ernie config for doctest (#19611) --- src/transformers/models/ernie/configuration_ernie.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/ernie/configuration_ernie.py b/src/transformers/models/ernie/configuration_ernie.py index 26a6ab535f..7fd1c861da 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 531274f395..76aec4a03a 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