tests/test_modeling_tf_common.py · 9e9f6b8a45de29d521a7b46bd6762d0eaaa6fa29 · chenpangpang / transformers

TF generate refactor - Sample (#15793) · baab5e7c

Joao Gante authored Mar 02, 2022



* Add TF logits wrappers 

* Add sample method

* add tests for TF logit wrappers

* TF generate sample tests now run on CPU
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

baab5e7c

test_modeling_tf_common.py 65.2 KB

Replace test_modeling_tf_common.py