Unverified Commit 82df83a9 authored by Ayaka's avatar Ayaka Committed by GitHub
Browse files

[Doctest] Add `configuration_plbart.py` (#19809)

Additionally, I updated the doctest format to be consistent with BERT.
parent 22502ebb
...@@ -91,16 +91,17 @@ class PLBartConfig(PretrainedConfig): ...@@ -91,16 +91,17 @@ class PLBartConfig(PretrainedConfig):
Example: Example:
```python ```python
>>> from transformers import PLBartModel, PLBartConfig >>> from transformers import PLBartConfig, PLBartModel
>>> # Initializing a PLBART uclanlp/plbart-base style configuration >>> # Initializing a PLBART uclanlp/plbart-base style configuration
>>> configuration = PLBartConfig() >>> 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) >>> model = PLBartModel(configuration)
>>> # Accessing the model configuration >>> # Accessing the model configuration
>>> configuration = model.config >>> configuration = model.config
```""" ```"""
model_type = "plbart" model_type = "plbart"
keys_to_ignore_at_inference = ["past_key_values"] keys_to_ignore_at_inference = ["past_key_values"]
attribute_map = {"num_attention_heads": "encoder_attention_heads", "hidden_size": "d_model"} attribute_map = {"num_attention_heads": "encoder_attention_heads", "hidden_size": "d_model"}
......
...@@ -105,6 +105,7 @@ src/transformers/models/pegasus/configuration_pegasus.py ...@@ -105,6 +105,7 @@ src/transformers/models/pegasus/configuration_pegasus.py
src/transformers/models/pegasus/modeling_pegasus.py src/transformers/models/pegasus/modeling_pegasus.py
src/transformers/models/pegasus_x/configuration_pegasus_x.py src/transformers/models/pegasus_x/configuration_pegasus_x.py
src/transformers/models/perceiver/modeling_perceiver.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/plbart/modeling_plbart.py
src/transformers/models/poolformer/modeling_poolformer.py src/transformers/models/poolformer/modeling_poolformer.py
src/transformers/models/realm/configuration_realm.py src/transformers/models/realm/configuration_realm.py
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment