src/transformers/generation_tf_logits_process.py · baab5e7cdf04c5b2cd209de4e9af6cb2c51a30d2 · 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

generation_tf_logits_process.py 17.1 KB

Replace generation_tf_logits_process.py