transformers/tests/generation
2023-02-01 21:50:38 +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 Fix tiny typo (#20841) 2022-12-20 03:17:59 -05:00
test_configuration_utils.py [GenerationConfig] add additional kwargs handling (#21269) 2023-01-24 19:04:42 +01:00
test_flax_logits_process.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_flax_utils.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_framework_agnostic.py Template for framework-agnostic tests (#21348) 2023-01-31 11:33:18 +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 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_tf_utils.py Template for framework-agnostic tests (#21348) 2023-01-31 11:33:18 +00:00
test_utils.py Generate: decoder-only models can generate with inputs_embeds (#21405) 2023-02-01 21:50:38 +00:00