diff --git a/src/transformers/models/bart/configuration_bart.py b/src/transformers/models/bart/configuration_bart.py index b7262f13fe3..2558311a44b 100644 --- a/src/transformers/models/bart/configuration_bart.py +++ b/src/transformers/models/bart/configuration_bart.py @@ -96,12 +96,12 @@ class BartConfig(PretrainedConfig): Example: ```python - >>> from transformers import BartModel, BartConfig + >>> from transformers import BartConfig, BartModel >>> # Initializing a BART facebook/bart-large style configuration >>> configuration = BartConfig() - >>> # Initializing a model from the facebook/bart-large style configuration + >>> # Initializing a model (with random weights) from the facebook/bart-large style configuration >>> model = BartModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 6c571f5aa2c..7f28f750725 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -14,6 +14,7 @@ docs/source/en/model_doc/encoder-decoder.mdx src/transformers/generation_utils.py src/transformers/models/albert/modeling_albert.py src/transformers/models/albert/modeling_tf_albert.py +src/transformers/models/bart/configuration_bart.py src/transformers/models/bart/modeling_bart.py src/transformers/models/beit/modeling_beit.py src/transformers/models/bert/configuration_bert.py