Sherman Siu 865da84abb Add Epsilon- and Eta-Sampling (#21121) ...

* Add epsilon- and eta-sampling.

Add epsilon- and eta-sampling, following the official code from https://github.com/john-hewitt/truncation-sampling and adapting to be more configurable, as required by Huggingface transformers.

* Add unit tests for epsilon- and eta-sampling.

* Black: fix code formatting.

* Fix docstring spacing.

* Clean up newlines.

* Fix implementation bugs and their associated tests.

* Remove epsilon- and eta-sampling parameters from PretrainedConfig.

* Clarify and clean up the documentation.

* Remove parameters for PretrainedConfig test.

2023-01-17 13:04:32 -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

Generate: use GenerationConfig as the basis for .generate() parametrization (#20388)

2022-12-15 18:27:20 +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 Epsilon- and Eta-Sampling (#21121)

2023-01-17 13:04:32 -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: Fix CI related to #20727 (#21003)

2023-01-04 20:26:56 +00:00