This commit is contained in:
Vasqu 2025-06-30 11:56:16 +02:00
parent 01227640ae
commit f46d6a48eb

View File

@ -809,6 +809,7 @@ class BertEncoderDecoderModelTest(EncoderDecoderMixin, unittest.TestCase):
encoder_config = config_and_inputs["config"]
decoder_config = config_and_inputs["decoder_config"]
encoder_config._attn_implementation = "eager"
decoder_config._attn_implementation = "eager"
encoder_config.position_embedding_type = "relative_key_query"
decoder_config.position_embedding_type = "relative_key_query"