From f8244014a5a7fafc82ead1d2930ccac777ce4527 Mon Sep 17 00:00:00 2001 From: Jing Hua <59118459+ztjhz@users.noreply.github.com> Date: Fri, 14 Oct 2022 22:45:37 +0800 Subject: [PATCH] Visual Bert config for doctest (#19605) --- .../models/visual_bert/configuration_visual_bert.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/visual_bert/configuration_visual_bert.py b/src/transformers/models/visual_bert/configuration_visual_bert.py index 60a3692644..e2f71119a5 100644 --- a/src/transformers/models/visual_bert/configuration_visual_bert.py +++ b/src/transformers/models/visual_bert/configuration_visual_bert.py @@ -96,12 +96,12 @@ class VisualBertConfig(PretrainedConfig): Example: ```python - >>> from transformers import VisualBertModel, VisualBertConfig + >>> from transformers import VisualBertConfig, VisualBertModel >>> # Initializing a VisualBERT visualbert-vqa-coco-pre style configuration >>> configuration = VisualBertConfig.from_pretrained("visualbert-vqa-coco-pre") - >>> # Initializing a model from the visualbert-vqa-coco-pre style configuration + >>> # Initializing a model (with random weights) from the visualbert-vqa-coco-pre style configuration >>> model = VisualBertModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index cb82c0024a..dd8fe5f00b 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -120,6 +120,7 @@ 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/visual_bert/configuration_visual_bert.py src/transformers/models/wav2vec2/modeling_wav2vec2.py src/transformers/models/wav2vec2/tokenization_wav2vec2.py src/transformers/models/wav2vec2_conformer/modeling_wav2vec2_conformer.py