tests/test_generation_utils.py · 8d6acc6c2931e4b41baf099e1101da1635474614 · chenpangpang / transformers

[Beam Search] Correct returned beam scores (#14654) · 8d6acc6c

Patrick von Platen authored Jan 24, 2022

* better

* save intermediate

* finish code

* up

* docs

* Apply suggestions from code review

* up

* add compute transition  beam scores function to model and make sure scores are correct with eos

* apply nicos comments

* Apply suggestions from code review

* another fix

8d6acc6c

test_generation_utils.py 87.4 KB

Replace test_generation_utils.py