Fix a Bug, trainer_seq2seq.py, in the else branch at Line 172, genera…

…tion_inputs should be a dict (huggingface#14546) * fix bug, trainer_seq2seq.py, Line 172, generation_inputs must be a dict before feeding into self.model.generation() * fix bug, trainer_seq2seq.py, Line 172, generation_inputs must be a dict before feeding into self.model.generation()
Albertobegue · Jan 27, 2022 · ca2d57f · ca2d57f
1 parent 12d1dd6
commit ca2d57f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/trainer_seq2seq.py b/src/transformers/trainer_seq2seq.py
@@ -169,7 +169,7 @@ def prediction_step(
             # very ugly hack to make it work
             generation_inputs["input_ids"] = generation_inputs.pop(self.tokenizer.model_input_names[0])
         else:
-            generation_inputs = inputs["input_ids"]
+            generation_inputs = {"input_ids": inputs["input_ids"]}
 
         generated_tokens = self.model.generate(
             **generation_inputs,