9 Commits

Author	SHA1	Message	Date
Joao Gante	dad5ca83b2	TF: Finalize unpack_inputs-related changes (#16499) ... * Add unpack_inputs to remaining models * removed kwargs to `call()` in TF models * fix TF T5 tests	2022-04-04 16:37:33 +01:00
Yih-Dar	2199382dfd	Use random_attention_mask for TF tests (#16517) ... * use random_attention_mask for TF tests * Fix for TFCLIP test (for now). Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>	2022-04-01 16:53:07 +02:00
Sylvain Gugger	4975002df5	Reorganize file utils (#16264) ... * Split file_utils in several submodules * Fixes * Add back more objects * More fixes * Who exactly decided to import that from there? * Second suggestion to code with code review * Revert wront move * Fix imports * Adapt all imports * Adapt all imports everywhere * Revert this import, will fix in a separate commit	2022-03-23 10:26:33 -04:00
Joao Gante	204c54d411	TF: add beam search tests (#16202)	2022-03-16 15:44:33 +00:00
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	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