From f4b386765d917695ff607bf769242668fa493762 Mon Sep 17 00:00:00 2001 From: Vishwas Date: Thu, 13 Oct 2022 15:29:02 +0530 Subject: [PATCH] [Doctest] Fixing doctest bert_generation configuration (#19558) * Added (with random weights) in the comment before model initialization line * Added configuration_bert_generation.py to utils/documentation_tests.txt Co-authored-by: vishwaspai --- .../models/bert_generation/configuration_bert_generation.py | 2 +- utils/documentation_tests.txt | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/models/bert_generation/configuration_bert_generation.py b/src/transformers/models/bert_generation/configuration_bert_generation.py index b4fa06afdf..ca6792bd0c 100644 --- a/src/transformers/models/bert_generation/configuration_bert_generation.py +++ b/src/transformers/models/bert_generation/configuration_bert_generation.py @@ -72,7 +72,7 @@ class BertGenerationConfig(PretrainedConfig): >>> # Initializing a BertGeneration config >>> configuration = BertGenerationConfig() - >>> # Initializing a model from the config + >>> # Initializing a model (with random weights) from the config >>> model = BertGenerationEncoder(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 8460bf1e8e..64017129cb 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -22,6 +22,7 @@ src/transformers/models/beit/modeling_beit.py src/transformers/models/bert/configuration_bert.py src/transformers/models/bert/modeling_bert.py src/transformers/models/bert/modeling_tf_bert.py +src/transformers/models/bert_generation/configuration_bert_generation.py src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py src/transformers/models/big_bird/modeling_big_bird.py src/transformers/models/blenderbot/modeling_blenderbot.py