From e64798296f5f300bf16972943d738f4e43ba5eb3 Mon Sep 17 00:00:00 2001 From: Partho Date: Fri, 14 Oct 2022 15:07:37 +0530 Subject: [PATCH] [Doctest] Swin Config for doctest (#19594) --- src/transformers/models/swin/configuration_swin.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/swin/configuration_swin.py b/src/transformers/models/swin/configuration_swin.py index b1d0ceb9bb..b169e49a57 100644 --- a/src/transformers/models/swin/configuration_swin.py +++ b/src/transformers/models/swin/configuration_swin.py @@ -87,12 +87,12 @@ class SwinConfig(PretrainedConfig): Example: ```python - >>> from transformers import SwinModel, SwinConfig + >>> from transformers import SwinConfig, SwinModel >>> # Initializing a Swin microsoft/swin-tiny-patch4-window7-224 style configuration >>> configuration = SwinConfig() - >>> # Initializing a model from the microsoft/swin-tiny-patch4-window7-224 style configuration + >>> # Initializing a model (with random weights) from the microsoft/swin-tiny-patch4-window7-224 style configuration >>> model = SwinModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index a0d17cd3ad..2410125761 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -96,6 +96,7 @@ src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.p 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/swin/configuration_swin.py src/transformers/models/swin/modeling_swin.py src/transformers/models/swinv2/configuration_swinv2.py src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py