Joao Gante c38c55f4fb Generate: store special token tensors under a unique variable name (#31980) ... * rename stuff * english; this one shouldn't be changed * add a _ to the new var names * musicgen * derp		2024-07-22 14:06:49 +01:00
..
__init__.py
test_beam_constraints.py
test_beam_search.py	Time to Say Goodbye, torch 1.7 and 1.8 (#22291)	2023-03-21 19:22:01 +01:00
test_configuration_utils.py	Avoid race condition (#31973)	2024-07-15 17:56:24 +02:00
test_flax_logits_process.py	Adding FlaxNoRepeatNGramLogitsProcessor (#29677)	2024-04-02 11:39:33 +02:00
test_flax_utils.py	Add support for beam search's num_return_sequencs flag in flax (#23082)	2023-05-03 10:50:34 -04:00
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	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
test_tf_utils.py	Revert workaround for TF safetensors loading (#30128)	2024-04-09 11:04:18 +01:00
test_utils.py	Generate: store special token tensors under a unique variable name (#31980)	2024-07-22 14:06:49 +01:00