[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:
committed by
GitHub
parent
6deac5c824
commit
909f07092a
@@ -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__(
|
||||||
|
|||||||
@@ -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 = {
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user