transformers/tests/generation
jiqing-feng 7f91f168a1
fix assisted decoding (#31401)
* fix assisted decoding

* check None

* fix typo

* fix _prepare_special_tokens

* fix style

* fix lint

* add tests for assisted decoding

* fix style

* fix tests check
2024-07-03 09:22:56 +01:00
..
__init__.py
test_beam_constraints.py
test_beam_search.py
test_configuration_utils.py
test_flax_logits_process.py Adding FlaxNoRepeatNGramLogitsProcessor (#29677) 2024-04-02 11:39:33 +02:00
test_flax_utils.py
test_framework_agnostic.py Generation: fix handling of special tokens (#31254) 2024-06-06 15:21:32 +05:00
test_logits_process.py Pass device in Logits Processor's init (#29804) 2024-06-04 10:19:19 +05:00
test_stopping_criteria.py Fix single letter stop strings (#31448) 2024-06-18 14:07:16 +01:00
test_streamers.py
test_tf_logits_process.py
test_tf_utils.py Revert workaround for TF safetensors loading (#30128) 2024-04-09 11:04:18 +01:00
test_utils.py fix assisted decoding (#31401) 2024-07-03 09:22:56 +01:00