From 82df83a96ba593a474e9d1887a4df542e61ae73a Mon Sep 17 00:00:00 2001 From: Ayaka Date: Mon, 24 Oct 2022 18:32:55 +0800 Subject: [PATCH] [Doctest] Add `configuration_plbart.py` (#19809) Additionally, I updated the doctest format to be consistent with BERT. --- src/transformers/models/plbart/configuration_plbart.py | 7 ++++--- utils/documentation_tests.txt | 1 + 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/transformers/models/plbart/configuration_plbart.py b/src/transformers/models/plbart/configuration_plbart.py index 94a96a675d..e3d6c7fbe9 100644 --- a/src/transformers/models/plbart/configuration_plbart.py +++ b/src/transformers/models/plbart/configuration_plbart.py @@ -91,16 +91,17 @@ class PLBartConfig(PretrainedConfig): Example: ```python - >>> from transformers import PLBartModel, PLBartConfig + >>> from transformers import PLBartConfig, PLBartModel >>> # Initializing a PLBART uclanlp/plbart-base style configuration >>> configuration = PLBartConfig() - >>> # Initializing a model from the uclanlp/plbart-base style configuration + + >>> # Initializing a model (with random weights) from the uclanlp/plbart-base style configuration >>> model = PLBartModel(configuration) + >>> # Accessing the model configuration >>> configuration = model.config ```""" - model_type = "plbart" keys_to_ignore_at_inference = ["past_key_values"] attribute_map = {"num_attention_heads": "encoder_attention_heads", "hidden_size": "d_model"} diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index af7065cf1e..713b6af2e4 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -105,6 +105,7 @@ src/transformers/models/pegasus/configuration_pegasus.py src/transformers/models/pegasus/modeling_pegasus.py src/transformers/models/pegasus_x/configuration_pegasus_x.py src/transformers/models/perceiver/modeling_perceiver.py +src/transformers/models/plbart/configuration_plbart.py src/transformers/models/plbart/modeling_plbart.py src/transformers/models/poolformer/modeling_poolformer.py src/transformers/models/realm/configuration_realm.py