Generate: missing generation config eos token setting in encoder-decoder tests (#29146)

2024-02-20 16:17:51 +00:00
parent 1c81132e80
commit 857fd8eaab
4 changed files with 8 additions and 0 deletions
--- a/tests/models/speech_encoder_decoder/test_modeling_speech_encoder_decoder.py
+++ b/tests/models/speech_encoder_decoder/test_modeling_speech_encoder_decoder.py
@@ -351,6 +351,8 @@ class EncoderDecoderMixin:
            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
+        if hasattr(enc_dec_model.generation_config, "eos_token_id"):
+            enc_dec_model.generation_config.eos_token_id = None

        inputs = input_values if input_features is None else input_features