Unverified Commit 909f0709 authored by Xabier Lahuerta Vazquez's avatar Xabier Lahuerta Vazquez Committed by GitHub
Browse files

[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
parent 6deac5c8
......@@ -91,19 +91,17 @@ class BigBirdConfig(PretrainedConfig):
Example:
```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 =
BigBirdConfig()
>>> # Initializing a model from the google/bigbird-roberta-base style configuration >>> model =
BigBirdModel(configuration)
>>> # Accessing the model configuration >>> configuration = model.config
"""
>>> # Accessing the model configuration
>>> configuration = model.config
```"""
model_type = "big_bird"
def __init__(
......
......@@ -109,19 +109,17 @@ class BigBirdPegasusConfig(PretrainedConfig):
Example:
```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 =
BigBirdPegasusConfig()
>>> # Initializing a model from the bigbird-pegasus-base style configuration >>> model =
BigBirdPegasusModel(configuration)
>>> # Accessing the model configuration >>> configuration = model.config
"""
>>> # Accessing the model configuration
>>> configuration = model.config
```"""
model_type = "bigbird_pegasus"
keys_to_ignore_at_inference = ["past_key_values"]
attribute_map = {
......
......@@ -23,7 +23,9 @@ src/transformers/models/bert/configuration_bert.py
src/transformers/models/bert/modeling_bert.py
src/transformers/models/bert/modeling_tf_bert.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/big_bird/configuration_big_bird.py
src/transformers/models/big_bird/modeling_big_bird.py
src/transformers/models/blenderbot/configuration_blenderbot.py
src/transformers/models/blenderbot/modeling_blenderbot.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