examples/flax/summarization/run_summarization_flax.py · 2e72bbab2cd169903b1e77b439718c1bdc5d50b2 · chenpangpang / transformers

"IMG/cpio/vscode:/vscode.git/clone" did not exist on "bb7e10d93e2548784ee321c152e7bd3ec3930eec"

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_summarization_flax.py 42.8 KB

Replace run_summarization_flax.py