"vscode:/vscode.git/clone" did not exist on "7495924007f9fcdc58f1014c2d3b369898d803b9"
Unverified Commit 041fdc4a authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[SpeechEncoderDecoder] Make sure no EOS is generated in test (#15655)

parent e314c19a
......@@ -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
......
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