Skip to content

GitLab

Unverified Commit efa4f5f0 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix (#13395)

parent 596bb85f
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
tests/test_modeling_speech_encoder_decoder.py
View file @ efa4f5f0
...@@ -388,8 +388,8 @@ class Wav2Vec2BertModelTest(EncoderDecoderMixin, unittest.TestCase): ...@@ -388,8 +388,8 @@ class Wav2Vec2BertModelTest(EncoderDecoderMixin, unittest.TestCase):
) )
def get_encoder_decoder_model(self, config, decoder_config): def get_encoder_decoder_model(self, config, decoder_config):
encoder_model = Wav2Vec2Model(config) encoder_model = Wav2Vec2Model(config).eval()
decoder_model = BertLMHeadModel(decoder_config) decoder_model = BertLMHeadModel(decoder_config).eval()
return encoder_model, decoder_model return encoder_model, decoder_model
def prepare_config_and_inputs(self): def prepare_config_and_inputs(self):
...@@ -439,8 +439,8 @@ class Speech2TextBertModelTest(EncoderDecoderMixin, unittest.TestCase): ...@@ -439,8 +439,8 @@ class Speech2TextBertModelTest(EncoderDecoderMixin, unittest.TestCase):
) )
def get_encoder_decoder_model(self, config, decoder_config): def get_encoder_decoder_model(self, config, decoder_config):
encoder_model = Speech2TextEncoder(config) encoder_model = Speech2TextEncoder(config).eval()
decoder_model = BertLMHeadModel(decoder_config) decoder_model = BertLMHeadModel(decoder_config).eval()
return encoder_model, decoder_model return encoder_model, decoder_model
def prepare_config_and_inputs(self): def prepare_config_and_inputs(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment