From ef5899bf34c4971ba4c8f78441c8b007d57c58e7 Mon Sep 17 00:00:00 2001 From: Partho Date: Thu, 13 Oct 2022 09:28:59 +0530 Subject: [PATCH] [Doctest] GPT2 Config for doctest (#19549) --- src/transformers/models/gpt2/configuration_gpt2.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/gpt2/configuration_gpt2.py b/src/transformers/models/gpt2/configuration_gpt2.py index 950fcd2f2c..fe9c711d73 100644 --- a/src/transformers/models/gpt2/configuration_gpt2.py +++ b/src/transformers/models/gpt2/configuration_gpt2.py @@ -117,12 +117,12 @@ class GPT2Config(PretrainedConfig): Example: ```python - >>> from transformers import GPT2Model, GPT2Config + >>> from transformers import GPT2Config, GPT2Model >>> # Initializing a GPT2 configuration >>> configuration = GPT2Config() - >>> # Initializing a model from the configuration + >>> # Initializing a model (with random weights) from the configuration >>> model = GPT2Model(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 9ae4db8ac8..e81bf7a7df 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -42,6 +42,7 @@ 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/glpn/modeling_glpn.py +src/transformers/models/gpt2/configuration_gpt2.py src/transformers/models/gpt2/modeling_gpt2.py src/transformers/models/gptj/modeling_gptj.py src/transformers/models/groupvit/modeling_groupvit.py