Joao Gante
3f43d824b9
TF generate refactor - Beam Search ( #16374 )
...
* refactor TF beam search
* refactored generate can now properly use attention masks
* add force bos/eos logit processors
2022-04-06 18:19:34 +01: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
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
