SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Doctest] Add configuration_plbart.py (#19809)

Browse Source 
Additionally, I updated the doctest format to be consistent with BERT.
This commit is contained in:
Ayaka
2022-10-24 18:32:55 +08:00
committed by GitHub
parent 22502ebb85
commit 82df83a96b
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/models/plbart/configuration_plbart.py
View File

@@ -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"}

1
utils/documentation_tests.txt
View File

@@ -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