Joao Gante 1eda4a4102 Generate: save generation config with the models' .save_pretrained() (#21264)

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