From 506355ca75519242ec27f38e11f69e3cbca69d77 Mon Sep 17 00:00:00 2001 From: Partho Date: Thu, 20 Oct 2022 13:49:06 +0530 Subject: [PATCH] [Doctest] SpeechToTextTransformer2 Config for doctest (#19756) --- .../models/speech_to_text_2/configuration_speech_to_text_2.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/speech_to_text_2/configuration_speech_to_text_2.py b/src/transformers/models/speech_to_text_2/configuration_speech_to_text_2.py index b7f1bc34a2..8825372911 100644 --- a/src/transformers/models/speech_to_text_2/configuration_speech_to_text_2.py +++ b/src/transformers/models/speech_to_text_2/configuration_speech_to_text_2.py @@ -79,12 +79,12 @@ class Speech2Text2Config(PretrainedConfig): Example: ```python - >>> from transformers import Speech2Text2ForCausalLM, Speech2Text2Config + >>> from transformers import Speech2Text2Config, Speech2Text2ForCausalLM >>> # Initializing a Speech2Text2 s2t_transformer_s style configuration >>> configuration = Speech2Text2Config() - >>> # Initializing a model from the s2t_transformer_s style configuration + >>> # Initializing a model (with random weights) from the s2t_transformer_s style configuration >>> model = Speech2Text2ForCausalLM(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 1699c9005c..028e489f27 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -123,6 +123,7 @@ src/transformers/models/sew_d/modeling_sew_d.py src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py src/transformers/models/speech_to_text/configuration_speech_to_text.py src/transformers/models/speech_to_text/modeling_speech_to_text.py +src/transformers/models/speech_to_text_2/configuration_speech_to_text_2.py src/transformers/models/speech_to_text_2/modeling_speech_to_text_2.py src/transformers/models/segformer/modeling_tf_segformer.py src/transformers/models/squeezebert/configuration_squeezebert.py