Nadav Timor 42b36d7395 Speculative decoding: Test the target distribution (to prevent issues like #32867) (#34553) ... * Update test_utils.py * formatting * Update test_utils.py * formatting * formatting * Update test_utils.py * formatting * Update test_utils.py * formatting * format * comments at standard positions		2024-11-22 16:02:37 +01:00
..
__init__.py
test_beam_constraints.py
test_beam_search.py
test_configuration_utils.py	Fix PushToHubMixin when pusing to a PR revision (#34090)	2024-10-11 15:06:15 +02:00
test_flax_logits_process.py	Adding FlaxNoRepeatNGramLogitsProcessor (#29677)	2024-04-02 11:39:33 +02:00
test_flax_utils.py	Fix CI (#34458)	2024-10-29 08:26:04 +01:00
test_framework_agnostic.py	Generation: fix handling of special tokens (#31254)	2024-06-06 15:21:32 +05:00
test_fsdp.py	Default synced_gpus to True when using FullyShardedDataParallel (#33483)	2024-10-10 14:09:04 -04:00
test_logits_process.py	Add SynthID (watermerking by Google DeepMind) (#34350)	2024-10-23 21:18:52 +01:00
test_stopping_criteria.py	Dynamic number of speculative tokens in order to accelerate speculative decoding (#33258)	2024-09-11 14:22:28 +02:00
test_streamers.py	Update all references to canonical models (#29001)	2024-02-16 08:16:58 +01:00
test_tf_logits_process.py	fix: multilingual midel convert to tflite get wrong token (#32079)	2024-08-27 11:44:09 +02:00
test_tf_utils.py	Revert workaround for TF safetensors loading (#30128)	2024-04-09 11:04:18 +01:00
test_utils.py	Speculative decoding: Test the target distribution (to prevent issues like #32867) (#34553)	2024-11-22 16:02:37 +01:00