tests/generation/test_generation_utils.py · 667b823b894700ec848c4e0bdcfc65630157ef82 · 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