From 0ae3ec5b9de71c266b574130ed640957863ad54f Mon Sep 17 00:00:00 2001
From: Partho <parthodas6176@gmail.com>
Date: Thu, 13 Oct 2022 15:37:14 +0530
Subject: [PATCH] [Doctest] Add `configuration_vit.py` (#19561)

* ViT Config for doctest
---
 src/transformers/models/vit/configuration_vit.py | 4 ++--
 utils/documentation_tests.txt                    | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/transformers/models/vit/configuration_vit.py b/src/transformers/models/vit/configuration_vit.py
index 546f0a3bcf..ec4f319f67 100644
--- a/src/transformers/models/vit/configuration_vit.py
+++ b/src/transformers/models/vit/configuration_vit.py
@@ -77,12 +77,12 @@ class ViTConfig(PretrainedConfig):
     Example:
 
     ```python
-    >>> from transformers import ViTModel, ViTConfig
+    >>> from transformers import ViTConfig, ViTModel
 
     >>> # Initializing a ViT vit-base-patch16-224 style configuration
     >>> configuration = ViTConfig()
 
-    >>> # Initializing a model from the vit-base-patch16-224 style configuration
+    >>> # Initializing a model (with random weights) from the vit-base-patch16-224 style configuration
     >>> model = ViTModel(configuration)
 
     >>> # Accessing the model configuration
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index f3c684239b..9ad5ca2ba7 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -99,6 +99,7 @@ src/transformers/models/van/modeling_van.py
 src/transformers/models/videomae/modeling_videomae.py
 src/transformers/models/vilt/modeling_vilt.py
 src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
+src/transformers/models/vit/configuration_vit.py
 src/transformers/models/vit/modeling_vit.py
 src/transformers/models/vit/modeling_tf_vit.py
 src/transformers/models/vit_mae/modeling_vit_mae.py