tests/generation/test_generation_logits_process.py · 3f43d824b909ec92cde5311e9be016767c0fb11b · chenpangpang / transformers

Joao Gante authored Apr 06, 2022

* refactor TF beam search

* refactored generate can now properly use attention masks

* add force bos/eos logit processors

3f43d824

test_generation_logits_process.py 22.3 KB