Motoki Wu 9960506cbe Fix multiple eos_token_ids in model.generate(...) (#21461) ... * add tests with multiple eos_token_ids * make math.prod instead of sum * make fixup * fix long and also use np.prod since math.prod does not exist <python 3.8 * make fixup * add prod util * use prod util instead of np.prod * make fixup * previous .long location * use tensor ops * remove prod * remove prod * update device * make fixup * fix none		2023-02-08 13:48:46 -05:00
..
__init__.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
test_beam_constraints.py	Generate: move generation_*.py src files into generation/*.py (#20096)	2022-11-09 15:34:08 +00:00
test_beam_search.py	Fix tiny typo (#20841)	2022-12-20 03:17:59 -05:00
test_configuration_utils.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
test_flax_logits_process.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
test_flax_utils.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
test_framework_agnostic.py	Generate: TF compute_transition_scores (#21341)	2023-02-08 16:36:43 +00:00
test_logits_process.py	Add hallucination filter (#18675)	2023-01-19 11:20:25 -05:00
test_stopping_criteria.py	Generate: move generation_*.py src files into generation/*.py (#20096)	2022-11-09 15:34:08 +00:00
test_tf_logits_process.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
test_tf_utils.py	Generate: TF can now generate from embeddings in encoder-decoder models (#21475)	2023-02-07 11:18:23 +00:00
test_utils.py	Fix multiple eos_token_ids in model.generate(...) (#21461)	2023-02-08 13:48:46 -05:00