From f3b3810fe6502a26bd51f187a2f8e1d41e295230 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Thu, 19 Sep 2024 11:55:26 +0100 Subject: [PATCH] rag: fix CI (#33578) --- tests/models/rag/test_modeling_rag.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/rag/test_modeling_rag.py b/tests/models/rag/test_modeling_rag.py index 392ff40d77..d00c063441 100644 --- a/tests/models/rag/test_modeling_rag.py +++ b/tests/models/rag/test_modeling_rag.py @@ -653,7 +653,7 @@ class RagDPRT5Test(RagTestMixin, unittest.TestCase): def config_and_inputs(self): question_encoder_tester = DPRModelTester(self) dpr_config_and_inputs = question_encoder_tester.prepare_config_and_inputs() - generator_tester = T5ModelTester(self, vocab_size=1100) + generator_tester = T5ModelTester(self, vocab_size=1101) t5_config_and_inputs = generator_tester.prepare_config_and_inputs() (question_encoder_config, input_ids, _, input_mask, _, _, _) = dpr_config_and_inputs