4 Commits

Author	SHA1	Message	Date
Matt	cd4c5c9060	TF XLA greedy generation (#15786) ... * First attempt at TF XLA generation * Fix comments * Update XLA greedy generate with direct XLA calls * Support attention mask, prepare_inputs_for_generation no longer hardcoded for greedy * Handle position_ids correctly * make xla generate work for non xla case * force using xla generate * refactor * more fixes * finish cleaning * finish * finish * clean gpt2 tests * add gpt2 tests * correct more cases * up * finish * finish * more fixes * flake 8 stuff * final rag fix * Update src/transformers/models/rag/modeling_tf_rag.py * finish t5 as well * finish * Update src/transformers/generation_utils.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>	2022-03-15 14:19:20 +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
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