SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

rag: fix CI (#33578)

Browse Source
This commit is contained in:
Joao Gante
2024-09-19 11:55:26 +01:00
committed by GitHub
parent d7975a5874
commit f3b3810fe6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/rag/test_modeling_rag.py
View File

@@ -653,7 +653,7 @@ class RagDPRT5Test(RagTestMixin, unittest.TestCase):
def config_and_inputs(self): def config_and_inputs(self):
question_encoder_tester = DPRModelTester(self) question_encoder_tester = DPRModelTester(self)
dpr_config_and_inputs = question_encoder_tester.prepare_config_and_inputs() 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() t5_config_and_inputs = generator_tester.prepare_config_and_inputs()
(question_encoder_config, input_ids, _, input_mask, _, _, _) = dpr_config_and_inputs (question_encoder_config, input_ids, _, input_mask, _, _, _) = dpr_config_and_inputs