From 9bd5d97cdd20e31c9372c464bc1e1a68ea2bc6d8 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 2 Sep 2021 18:47:09 +0200 Subject: [PATCH] up (#13396) --- tests/test_modeling_speech_encoder_decoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_speech_encoder_decoder.py b/tests/test_modeling_speech_encoder_decoder.py index 453adbf73ba..e5562d4c681 100644 --- a/tests/test_modeling_speech_encoder_decoder.py +++ b/tests/test_modeling_speech_encoder_decoder.py @@ -493,8 +493,8 @@ class Speech2TextBertModelTest(EncoderDecoderMixin, unittest.TestCase): @require_torch class Wav2Vec2Speech2Text2(EncoderDecoderMixin, unittest.TestCase): def get_encoder_decoder_model(self, config, decoder_config): - encoder_model = Wav2Vec2Model(config) - decoder_model = Speech2Text2ForCausalLM(decoder_config) + encoder_model = Wav2Vec2Model(config).eval() + decoder_model = Speech2Text2ForCausalLM(decoder_config).eval() return encoder_model, decoder_model def prepare_config_and_inputs(self):