From fd99ce3329ca63372332e615692f48802e6eb45e Mon Sep 17 00:00:00 2001 From: JuanCarlosPi Date: Tue, 18 Oct 2022 07:57:34 -0500 Subject: [PATCH] [Doctest] Add configuration_wav2vec2.py to documentation_tests.py (#19698) --- src/transformers/models/wav2vec2/configuration_wav2vec2.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/wav2vec2/configuration_wav2vec2.py b/src/transformers/models/wav2vec2/configuration_wav2vec2.py index 6b96d9fc3f..ab7d116118 100644 --- a/src/transformers/models/wav2vec2/configuration_wav2vec2.py +++ b/src/transformers/models/wav2vec2/configuration_wav2vec2.py @@ -184,12 +184,12 @@ class Wav2Vec2Config(PretrainedConfig): Example: ```python - >>> from transformers import Wav2Vec2Model, Wav2Vec2Config + >>> from transformers import Wav2Vec2Config, Wav2Vec2Model >>> # Initializing a Wav2Vec2 facebook/wav2vec2-base-960h style configuration >>> configuration = Wav2Vec2Config() - >>> # Initializing a model from the facebook/wav2vec2-base-960h style configuration + >>> # Initializing a model (with random weights) from the facebook/wav2vec2-base-960h style configuration >>> model = Wav2Vec2Model(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 23183ea687..835837bc38 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -138,6 +138,7 @@ 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/configuration_wav2vec2.py src/transformers/models/wav2vec2/modeling_wav2vec2.py src/transformers/models/wav2vec2/tokenization_wav2vec2.py src/transformers/models/wav2vec2_conformer/modeling_wav2vec2_conformer.py