From 041fdc4a7ed4762c2adf791f46b5e8769766c4d4 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 15 Feb 2022 09:13:55 +0100 Subject: [PATCH] [SpeechEncoderDecoder] Make sure no EOS is generated in test (#15655) --- tests/test_modeling_speech_encoder_decoder.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_modeling_speech_encoder_decoder.py b/tests/test_modeling_speech_encoder_decoder.py index 6b0e5cf12f..6a5f1b589c 100644 --- a/tests/test_modeling_speech_encoder_decoder.py +++ b/tests/test_modeling_speech_encoder_decoder.py @@ -308,6 +308,11 @@ class EncoderDecoderMixin: enc_dec_model = SpeechEncoderDecoderModel(encoder=encoder_model, decoder=decoder_model) enc_dec_model.to(torch_device) + # make sure EOS token is set to None to prevent early stopping of generation + enc_dec_model.config.eos_token_id = None + if hasattr(enc_dec_model.config, "decoder") and hasattr(enc_dec_model.config.decoder, "eos_token_id"): + enc_dec_model.config.decoder.eos_token_id = None + inputs = input_values if input_features is None else input_features # Bert does not have a bos token id, so use pad_token_id instead