tests/gpt2/test_modeling_tf_gpt2.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_modeling_tf_gpt2.py 23.5 KB