[Doctest] Add configuration_bigbird_pegasus.py and configuration_big_bird.py (#19606)

* [Doctest] Add `configuration_bigbird_pegasus.py` and `configuration_big_bird`

[Doctest] Re-style `configuration_big_bird.py`

* [Doctest] One python instruction per line

* [Doctest] Fix styling

* [Doctest] More styling fixes
This commit is contained in:
Xabier Lahuerta Vazquez
2022-10-14 15:17:36 +02:00
committed by GitHub
parent 6deac5c824
commit 909f07092a
3 changed files with 18 additions and 20 deletions

View File

@@ -91,19 +91,17 @@ class BigBirdConfig(PretrainedConfig):
Example: Example:
```python ```python
>>> from transformers import BigBirdConfig, BigBirdModel
``` >>> # Initializing a BigBird google/bigbird-roberta-base style configuration
>>> configuration = BigBirdConfig()
>>> from transformers import BigBirdModel, BigBirdConfig >>> # Initializing a model (with random weights) from the google/bigbird-roberta-base style configuration
>>> model = BigBirdModel(configuration)
>>> # Initializing a BigBird google/bigbird-roberta-base style configuration >>> configuration = >>> # Accessing the model configuration
BigBirdConfig() >>> configuration = model.config
```"""
>>> # Initializing a model from the google/bigbird-roberta-base style configuration >>> model =
BigBirdModel(configuration)
>>> # Accessing the model configuration >>> configuration = model.config
"""
model_type = "big_bird" model_type = "big_bird"
def __init__( def __init__(

View File

@@ -109,19 +109,17 @@ class BigBirdPegasusConfig(PretrainedConfig):
Example: Example:
```python ```python
>>> from transformers import BigBirdPegasusConfig, BigBirdPegasusModel
``` >>> # Initializing a BigBirdPegasus bigbird-pegasus-base style configuration
>>> configuration = BigBirdPegasusConfig()
>>> from transformers import BigBirdPegasusModel, BigBirdPegasusConfig >>> # Initializing a model (with random weights) from the bigbird-pegasus-base style configuration
>>> model = BigBirdPegasusModel(configuration)
>>> # Initializing a BigBirdPegasus bigbird-pegasus-base style configuration >>> configuration = >>> # Accessing the model configuration
BigBirdPegasusConfig() >>> configuration = model.config
```"""
>>> # Initializing a model from the bigbird-pegasus-base style configuration >>> model =
BigBirdPegasusModel(configuration)
>>> # Accessing the model configuration >>> configuration = model.config
"""
model_type = "bigbird_pegasus" model_type = "bigbird_pegasus"
keys_to_ignore_at_inference = ["past_key_values"] keys_to_ignore_at_inference = ["past_key_values"]
attribute_map = { attribute_map = {

View File

@@ -23,7 +23,9 @@ src/transformers/models/bert/configuration_bert.py
src/transformers/models/bert/modeling_bert.py src/transformers/models/bert/modeling_bert.py
src/transformers/models/bert/modeling_tf_bert.py src/transformers/models/bert/modeling_tf_bert.py
src/transformers/models/bert_generation/configuration_bert_generation.py src/transformers/models/bert_generation/configuration_bert_generation.py
src/transformers/models/bigbird_pegasus/configuration_bigbird_pegasus.py
src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py
src/transformers/models/big_bird/configuration_big_bird.py
src/transformers/models/big_bird/modeling_big_bird.py src/transformers/models/big_bird/modeling_big_bird.py
src/transformers/models/blenderbot/configuration_blenderbot.py src/transformers/models/blenderbot/configuration_blenderbot.py
src/transformers/models/blenderbot/modeling_blenderbot.py src/transformers/models/blenderbot/modeling_blenderbot.py