From 123f65eea644f59fff65704a61ce8488d25569a8 Mon Sep 17 00:00:00 2001 From: Partho Date: Thu, 20 Oct 2022 13:46:39 +0530 Subject: [PATCH] [Doctest] SqueezeBERT Config for doctest (#19758) --- .../models/squeezebert/configuration_squeezebert.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/squeezebert/configuration_squeezebert.py b/src/transformers/models/squeezebert/configuration_squeezebert.py index 41b47ff575..639be83a6c 100644 --- a/src/transformers/models/squeezebert/configuration_squeezebert.py +++ b/src/transformers/models/squeezebert/configuration_squeezebert.py @@ -94,12 +94,12 @@ class SqueezeBertConfig(PretrainedConfig): Examples: ```python - >>> from transformers import SqueezeBertModel, SqueezeBertConfig + >>> from transformers import SqueezeBertConfig, SqueezeBertModel >>> # Initializing a SqueezeBERT configuration >>> configuration = SqueezeBertConfig() - >>> # Initializing a model from the configuration above + >>> # Initializing a model (with random weights) from the configuration above >>> model = SqueezeBertModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index df2be97980..1699c9005c 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -125,6 +125,7 @@ src/transformers/models/speech_to_text/configuration_speech_to_text.py src/transformers/models/speech_to_text/modeling_speech_to_text.py src/transformers/models/speech_to_text_2/modeling_speech_to_text_2.py src/transformers/models/segformer/modeling_tf_segformer.py +src/transformers/models/squeezebert/configuration_squeezebert.py src/transformers/models/swin/configuration_swin.py src/transformers/models/swin/modeling_swin.py src/transformers/models/swinv2/configuration_swinv2.py