From 585f9c6d9efa9f6e93888b6adf84912ba3f98dfc Mon Sep 17 00:00:00 2001 From: Partho Date: Fri, 14 Oct 2022 20:52:29 +0530 Subject: [PATCH] [Doctest] DistilBERT Config for doctest (#19621) --- .../models/distilbert/configuration_distilbert.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/distilbert/configuration_distilbert.py b/src/transformers/models/distilbert/configuration_distilbert.py index c746ad0d64..b36917bc07 100644 --- a/src/transformers/models/distilbert/configuration_distilbert.py +++ b/src/transformers/models/distilbert/configuration_distilbert.py @@ -87,12 +87,12 @@ class DistilBertConfig(PretrainedConfig): Examples: ```python - >>> from transformers import DistilBertModel, DistilBertConfig + >>> from transformers import DistilBertConfig, DistilBertModel >>> # Initializing a DistilBERT configuration >>> configuration = DistilBertConfig() - >>> # Initializing a model from the configuration + >>> # Initializing a model (with random weights) from the configuration >>> model = DistilBertModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 7935a743c4..3dbe5aee20 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -47,6 +47,7 @@ src/transformers/models/deit/configuration_deit.py src/transformers/models/deit/modeling_deit.py src/transformers/models/deit/modeling_tf_deit.py src/transformers/models/detr/modeling_detr.py +src/transformers/models/distilbert/configuration_distilbert.py src/transformers/models/dpt/modeling_dpt.py src/transformers/models/electra/modeling_electra.py src/transformers/models/electra/modeling_tf_electra.py