Joao Gante 70d5df6107 Generate: unify LogitsWarper and LogitsProcessor (#32626)

2024-08-16 11:20:41 +01:00

..

__init__.py

…

test_beam_constraints.py

…

test_beam_search.py

…

test_configuration_utils.py

Follow up for #31973 (#32025)

2024-07-25 16:12:23 +02:00

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

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

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

Generate: unify LogitsWarper and LogitsProcessor (#32626)

2024-08-16 11:20:41 +01:00