From 4d77f18cba36b9e275b7307dc8f8f2de7bac7374 Mon Sep 17 00:00:00 2001 From: Partho Date: Mon, 17 Oct 2022 22:09:15 +0530 Subject: [PATCH] [Doctest] Data2VecAudio Config for doctest (#19635) --- .../models/data2vec/configuration_data2vec_audio.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/data2vec/configuration_data2vec_audio.py b/src/transformers/models/data2vec/configuration_data2vec_audio.py index cc32f2cc69..c1aee46832 100644 --- a/src/transformers/models/data2vec/configuration_data2vec_audio.py +++ b/src/transformers/models/data2vec/configuration_data2vec_audio.py @@ -152,12 +152,12 @@ class Data2VecAudioConfig(PretrainedConfig): Example: ```python - >>> from transformers import Data2VecAudioModel, Data2VecAudioConfig + >>> from transformers import Data2VecAudioConfig, Data2VecAudioModel >>> # Initializing a Data2VecAudio facebook/data2vec-audio-base-960h style configuration >>> configuration = Data2VecAudioConfig() - >>> # Initializing a model from the facebook/data2vec-audio-base-960h style configuration + >>> # Initializing a model (with random weights) from the facebook/data2vec-audio-base-960h style configuration >>> model = Data2VecAudioModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 582f5a1355..980509feb2 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -40,6 +40,7 @@ src/transformers/models/convnext/modeling_convnext.py src/transformers/models/ctrl/configuration_ctrl.py src/transformers/models/ctrl/modeling_ctrl.py src/transformers/models/cvt/modeling_cvt.py +src/transformers/models/data2vec/configuration_data2vec_audio.py src/transformers/models/data2vec/configuration_data2vec_text.py src/transformers/models/data2vec/modeling_data2vec_audio.py src/transformers/models/data2vec/modeling_data2vec_vision.py