From 6cd8676cf358e100247921a54f8418cdec1b09a0 Mon Sep 17 00:00:00 2001 From: Partho Date: Thu, 13 Oct 2022 00:08:13 +0530 Subject: [PATCH] [Doctest] Beit Config for doctest (#19542) --- src/transformers/models/beit/configuration_beit.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/beit/configuration_beit.py b/src/transformers/models/beit/configuration_beit.py index e4cdb1eda2..ad6b9e864d 100644 --- a/src/transformers/models/beit/configuration_beit.py +++ b/src/transformers/models/beit/configuration_beit.py @@ -104,12 +104,12 @@ class BeitConfig(PretrainedConfig): Example: ```python - >>> from transformers import BeitModel, BeitConfig + >>> from transformers import BeitConfig, BeitModel >>> # Initializing a BEiT beit-base-patch16-224-pt22k style configuration >>> configuration = BeitConfig() - >>> # Initializing a model from the beit-base-patch16-224-pt22k style configuration + >>> # Initializing a model (with random weights) from the beit-base-patch16-224-pt22k style configuration >>> model = BeitModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 54a76c5b41..9ae4db8ac8 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -17,6 +17,7 @@ src/transformers/models/albert/modeling_albert.py src/transformers/models/albert/modeling_tf_albert.py src/transformers/models/bart/configuration_bart.py src/transformers/models/bart/modeling_bart.py +src/transformers/models/beit/configuration_beit.py src/transformers/models/beit/modeling_beit.py src/transformers/models/bert/configuration_bert.py src/transformers/models/bert/modeling_bert.py