From 7180e1725635e0a02442e47c3ef48ae32d8e72d5 Mon Sep 17 00:00:00 2001
From: KaranGarg <95518516+grgkaran03@users.noreply.github.com>
Date: Thu, 13 Oct 2022 18:35:02 +0530
Subject: [PATCH] [Doctests] Config files for `ViTMAE` and `YOSO` (#19567)

---
 src/transformers/models/vit_mae/configuration_vit_mae.py | 4 ++--
 src/transformers/models/yoso/configuration_yoso.py       | 4 ++--
 utils/documentation_tests.txt                            | 2 ++
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/transformers/models/vit_mae/configuration_vit_mae.py b/src/transformers/models/vit_mae/configuration_vit_mae.py
index cc8314e7a4..94a8c4c66a 100644
--- a/src/transformers/models/vit_mae/configuration_vit_mae.py
+++ b/src/transformers/models/vit_mae/configuration_vit_mae.py
@@ -82,12 +82,12 @@ class ViTMAEConfig(PretrainedConfig):
     Example:
 
     ```python
-    >>> from transformers import ViTMAEModel, ViTMAEConfig
+    >>> from transformers import ViTMAEConfig, ViTMAEModel
 
     >>> # Initializing a ViT MAE vit-mae-base style configuration
     >>> configuration = ViTMAEConfig()
 
-    >>> # Initializing a model from the vit-mae-base style configuration
+    >>> # Initializing a model (with random weights) from the vit-mae-base style configuration
     >>> model = ViTMAEModel(configuration)
 
     >>> # Accessing the model configuration
diff --git a/src/transformers/models/yoso/configuration_yoso.py b/src/transformers/models/yoso/configuration_yoso.py
index 8d98e278b8..7a2458146c 100644
--- a/src/transformers/models/yoso/configuration_yoso.py
+++ b/src/transformers/models/yoso/configuration_yoso.py
@@ -83,12 +83,12 @@ class YosoConfig(PretrainedConfig):
     Example:
 
     ```python
-    >>> from transformers import YosoModel, YosoConfig
+    >>> from transformers import YosoConfig, YosoModel
 
     >>> # Initializing a YOSO uw-madison/yoso-4096 style configuration
     >>> configuration = YosoConfig()
 
-    >>> # Initializing a model from the uw-madison/yoso-4096 style configuration
+    >>> # Initializing a model (with random weights) from the uw-madison/yoso-4096 style configuration
     >>> model = YosoModel(configuration)
 
     >>> # Accessing the model configuration
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index 71055664bb..d48073a5c3 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -106,6 +106,7 @@ 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
+src/transformers/models/vit_mae/configuration_vit_mae.py
 src/transformers/models/vit_msn/modeling_vit_msn.py
 src/transformers/models/wav2vec2/modeling_wav2vec2.py
 src/transformers/models/wav2vec2/tokenization_wav2vec2.py
@@ -118,3 +119,4 @@ src/transformers/models/whisper/modeling_tf_whisper.py
 src/transformers/models/yolos/configuration_yolos.py
 src/transformers/models/yolos/modeling_yolos.py
 src/transformers/models/x_clip/modeling_x_clip.py
+src/transformers/models/yoso/configuration_yoso.py