From fac1f4b1884e7a2ea2077b10fda68e4d57cee782 Mon Sep 17 00:00:00 2001 From: Mukesh K Date: Tue, 18 Oct 2022 19:08:57 +0530 Subject: [PATCH] ]Fixed pegasus config doctest (#19722) Co-authored-by: mukesh663 --- src/transformers/models/pegasus/configuration_pegasus.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/pegasus/configuration_pegasus.py b/src/transformers/models/pegasus/configuration_pegasus.py index ae5f8f0075..59dee43fea 100644 --- a/src/transformers/models/pegasus/configuration_pegasus.py +++ b/src/transformers/models/pegasus/configuration_pegasus.py @@ -88,12 +88,12 @@ class PegasusConfig(PretrainedConfig): Example: ```python - >>> from transformers import PegasusModel, PegasusConfig + >>> from transformers import PegasusConfig, PegasusModel >>> # Initializing a PEGASUS google/pegasus-large style configuration >>> configuration = PegasusConfig() - >>> # Initializing a model from the google/pegasus-large style configuration + >>> # Initializing a model (with random weights) from the google/pegasus-large style configuration >>> model = PegasusModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 6a5f2a8bf9..df7f01fe21 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -89,6 +89,7 @@ src/transformers/models/mobilevit/modeling_tf_mobilevit.py src/transformers/models/opt/modeling_opt.py src/transformers/models/opt/modeling_tf_opt.py src/transformers/models/owlvit/modeling_owlvit.py +src/transformers/models/pegasus/configuration_pegasus.py src/transformers/models/pegasus/modeling_pegasus.py src/transformers/models/perceiver/modeling_perceiver.py src/transformers/models/plbart/modeling_plbart.py