From e06bb34a4791a6ea8f40b928e3262f09d77115ac Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Wed, 18 Sep 2024 18:35:59 +0000 Subject: [PATCH] fix rag --- 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 392ff40d7702c8..d00c06344118b3 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