tests/generation/test_generation_utils.py · ef9c3ca348cb11c4ba951f8bdab068ecd8b847ce · chenpangpang / transformers

[Bug Fix] Beam search example in docs fails & a fix (integrating `max_length`... · ef9c3ca3

Chan Woo Kim authored Mar 07, 2022

[Bug Fix] Beam search example in docs fails & a fix (integrating `max_length` in `BeamScorer.finalize()`) (#15555)

* added the test and fix

* had left out a comment

ef9c3ca3

test_generation_utils.py 108 KB

Replace test_generation_utils.py