Joao Gante
70203b5937
TF generate refactor - past without encoder outputs ( #15944 )
...
* Remove packed past from generation_tf_utils
* update models with the new past format
* update template accordingly
2022-03-08 14:46:44 +00:00
Joao Gante
baab5e7cdf
TF generate refactor - Sample ( #15793 )
...
* 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>
2022-03-02 16:13:54 +00:00
Joao Gante
8a133490bf
Add TF generate sample tests with all logit processors ( #15852 )
...
* Add GPT2 TF generate sample test with all logits processor
* Add T5 generate sample test
2022-03-02 09:48:11 +00:00
Lysandre Debut
29c10a41d0
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
...
* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>
2022-02-23 15:46:28 -05:00