examples/pytorch/translation/run_translation.py · 2e72bbab2cd169903b1e77b439718c1bdc5d50b2 · chenpangpang / transformers

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

Matt authored Nov 15, 2023



* 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>

2e72bbab

run_translation.py 29.4 KB

Replace run_translation.py