From 4a3578f23fbce1580f420be1ff21a8c6431e839b Mon Sep 17 00:00:00 2001 From: Partho Date: Thu, 13 Oct 2022 15:32:40 +0530 Subject: [PATCH] [Doctest] DeiT Config for doctest (#19560) --- src/transformers/models/deit/configuration_deit.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/deit/configuration_deit.py b/src/transformers/models/deit/configuration_deit.py index f937ae8451..39faa9a110 100644 --- a/src/transformers/models/deit/configuration_deit.py +++ b/src/transformers/models/deit/configuration_deit.py @@ -80,12 +80,12 @@ class DeiTConfig(PretrainedConfig): Example: ```python - >>> from transformers import DeiTModel, DeiTConfig + >>> from transformers import DeiTConfig, DeiTModel >>> # Initializing a DeiT deit-base-distilled-patch16-224 style configuration >>> configuration = DeiTConfig() - >>> # Initializing a model from the deit-base-distilled-patch16-224 style configuration + >>> # Initializing a model (with random weights) from the deit-base-distilled-patch16-224 style configuration >>> model = DeiTModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 64017129cb..63fe29d6db 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -36,6 +36,7 @@ src/transformers/models/data2vec/modeling_data2vec_vision.py src/transformers/models/deberta/modeling_deberta.py src/transformers/models/deberta_v2/modeling_deberta_v2.py src/transformers/models/deformable_detr/modeling_deformable_detr.py +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