Incorrect setting for num_beams in translation and summarization examples (#27519)

* Remove the torch main_process_first context manager from TF examples

* Correctly set num_beams=1 in our examples, and add a guard in GenerationConfig.validate()

* Update src/transformers/generation/configuration_utils.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
This commit is contained in:
Matt
2023-11-15 18:18:54 +00:00
committed by GitHub
parent e6522e49a7
commit 2e72bbab2c
7 changed files with 53 additions and 55 deletions

View File

@@ -249,7 +249,7 @@ class DataTrainingArguments:
},
)
num_beams: Optional[int] = field(
default=None,
default=1,
metadata={
"help": (
"Number of beams to use for evaluation. This argument will be passed to ``model.generate``, "