Skip to content

GitLab

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

up (#13396)

parent efa4f5f0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_modeling_speech_encoder_decoder.py
View file @ 9bd5d97c
...@@ -493,8 +493,8 @@ class Speech2TextBertModelTest(EncoderDecoderMixin, unittest.TestCase): ...@@ -493,8 +493,8 @@ class Speech2TextBertModelTest(EncoderDecoderMixin, unittest.TestCase):
@require_torch @require_torch
class Wav2Vec2Speech2Text2(EncoderDecoderMixin, unittest.TestCase): class Wav2Vec2Speech2Text2(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 = Speech2Text2ForCausalLM(decoder_config) decoder_model = Speech2Text2ForCausalLM(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