transformers/tests/generation
2025-04-22 11:07:34 +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 Allow Exclusion of Input IDs from RepetitionPenaltyLogitsProcessor (#37625) 2025-04-21 15:46:05 +01: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 [test] update test_past_key_values_format (#37614) 2025-04-22 11:07:34 +01:00