Arthur 94a7edd938 [GenerationConfig] add additional kwargs handling (#21269) ...

* add additional kwargs handling

* fix issue when serializing

* correct order of kwargs removal for serialization in from dict

* add `dict_torch_dtype_to_str` in case a dtype is needed for generation

* add condition when adding the kwargs : not from config

* Add comment based on review

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* add test function

* default None when poping arg

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

2023-01-24 19:04:42 +01: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_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