Jonathan Mamou 121f91d36c prune LM Head for USD (#36695) ... * initial commit * fix * fix style * set default to prune * add tests * comment * remove prune flag from generate * address Joao's comments * deprecate_kwarg * add doc * fix target_vocab_size * Update src/transformers/generation/candidate_generator.py Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com> * Update src/transformers/generation/candidate_generator.py Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com> * Update src/transformers/generation/candidate_generator.py Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com> * Update src/transformers/generation/candidate_generator.py Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com> * fix deprecated argument assistant_model_device --------- Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>		2025-04-08 16:44:10 +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	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_beam_search.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_candidate_generator.py	prune LM Head for USD (#36695)	2025-04-08 16:44:10 +01:00
test_configuration_utils.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_fsdp.py	[MLU] Fix FA2 check error, remove deepspeed-mlu deps. (#36159)	2025-03-31 11:02:49 +02:00
test_logits_process.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_stopping_criteria.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_streamers.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_utils.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00