enable 5 cases on XPU (#37507)

* make speecht5 test_batch_generation pass on XPU

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

* enable 4 GlmIntegrationTest cases on XPU

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

* fix style

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

* Update src/transformers/testing_utils.py

Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>

---------

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
This commit is contained in:
Yao Matrix
2025-04-16 15:28:02 +08:00
committed by GitHub
parent 3165eb7c28
commit 5ab7a7c640
3 changed files with 16 additions and 2 deletions

View File

@@ -22,7 +22,7 @@ from transformers.testing_utils import (
is_flaky,
require_flash_attn,
require_torch,
require_torch_large_gpu,
require_torch_large_accelerator,
require_torch_sdpa,
slow,
torch_device,
@@ -309,7 +309,7 @@ class GlmModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin,
@slow
@require_torch_large_gpu
@require_torch_large_accelerator
class GlmIntegrationTest(unittest.TestCase):
input_text = ["Hello I am doing", "Hi today"]
model_id = "THUDM/glm-4-9b"