SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix (#13395)

Browse Source
This commit is contained in:
Patrick von Platen
2021-09-02 18:11:26 +02:00
committed by GitHub
parent 596bb85f2f
commit efa4f5f0ea
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_modeling_speech_encoder_decoder.py
View File

@@ -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):
) )
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):