transformers/tests/generation
Yao Matrix 89542fb81c
enable more test cases on xpu (#38572)
* enable glm4 integration cases on XPU, set xpu expectation for blip2

Signed-off-by: Matrix YAO <matrix.yao@intel.com>

* more

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* fix style

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* refine wording

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* refine test case names

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* run

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* add gemma2 and chameleon

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* fix review comments

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

---------

Signed-off-by: Matrix YAO <matrix.yao@intel.com>
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
2025-06-06 09:29:51 +02: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 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_fsdp.py enable generation fsdp/utils cases on XPU (#38009) 2025-05-09 20:52:41 +00:00
test_logits_process.py Allow Exclusion of Input IDs from RepetitionPenaltyLogitsProcessor (#37625) 2025-04-21 15:46:05 +01:00
test_paged_attention.py enable more test cases on xpu (#38572) 2025-06-06 09:29:51 +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 🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288) 2025-05-23 17:17:38 +02:00