transformers/tests/generation
2023-01-23 16:21:44 +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 Generate: save generation config with the models' .save_pretrained() (#21264) 2023-01-23 16:21:44 +00: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_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 Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_utils.py Generate: documented function to compute the transition scores (#21191) 2023-01-20 12:50:01 +00:00