riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 13:20:12 +06:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

enable csm integration cases on xpu, all passed (#38140)

Browse Source 
* enable csm test cases on XPU, all passed

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

* fix style

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

---------

Signed-off-by: Matrix Yao <matrix.yao@intel.com>
This commit is contained in:
Yao Matrix 2025-05-15 15:46:29 +08:00 committed by GitHub
parent e5a48785d9
commit 0173a99e73
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/models/csm/test_modeling_csm.py
View File

@ -30,7 +30,7 @@ from transformers import (
)
from transformers.testing_utils import (
cleanup,
require_torch_gpu,
require_torch_accelerator,
slow,
torch_device,
)
@ -430,7 +430,7 @@ class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
return ds[0]
@slow
@require_torch_gpu
@require_torch_accelerator
def test_1b_model_integration_generate(self):
"""
Tests the generated tokens match the ones from the original model implementation.
@ -474,7 +474,7 @@ class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
torch.testing.assert_close(output_tokens.cpu(), EXPECTED_OUTPUT_TOKENS)
@slow
@require_torch_gpu
@require_torch_accelerator
def test_1b_model_integration_generate_no_audio(self):
"""
Tests the generated tokens match the ones from the original model implementation.
@ -535,7 +535,7 @@ class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
torch.testing.assert_close(output_tokens.cpu(), EXPECTED_OUTPUT_TOKENS)
@slow
@require_torch_gpu
@require_torch_accelerator
def test_1b_model_integration_generate_multiple_audio(self):
"""
Test the generated tokens match the ones from the original model implementation.
@ -594,7 +594,7 @@ class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
torch.testing.assert_close(output_tokens.cpu(), EXPECTED_OUTPUT_TOKENS)
@slow
@require_torch_gpu
@require_torch_accelerator
def test_1b_model_integration_generate_batched(self):
"""
Test the generated tokens match the ones from the original model implementation.