Xin Qiu 453079c7f8 🚨🚨 Fix beam score calculation issue for decoder-only models (#27351) ... * Fix beam score calculation issue for decoder-only models * Update beam search test and fix code quality issue * Fix beam_sample, group_beam_search and constrained_beam_search * Split test for pytorch and TF, add documentation --------- Co-authored-by: Xin Qiu <xin.qiu@sentient.ai>		2023-11-15 12:49:14 +00: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	Time to Say Goodbye, torch 1.7 and 1.8 (#22291)	2023-03-21 19:22:01 +01:00
test_configuration_utils.py	Generation: strict generation config validation at save time (#25411)	2023-08-10 10:42:34 +01:00
test_flax_logits_process.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
test_flax_utils.py	Add support for beam search's num_return_sequencs flag in flax (#23082)	2023-05-03 10:50:34 -04:00
test_framework_agnostic.py	🚨🚨 Fix beam score calculation issue for decoder-only models (#27351)	2023-11-15 12:49:14 +00:00
test_logits_process.py	Add early stopping for Bark generation via logits processor (#26675)	2023-10-27 11:07:33 +01:00
test_stopping_criteria.py	Generate: move generation_*.py src files into generation/*.py (#20096)	2022-11-09 15:34:08 +00:00
test_streamers.py	Generate: TextIteratorStreamer timeout (#22576)	2023-04-05 09:57:46 +01:00
test_tf_logits_process.py	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
test_tf_utils.py	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
test_utils.py	[CircleCI] skip test_assisted_decoding_sample for everyone (#27511)	2023-11-15 10:17:51 +01:00