SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

enable more test cases on xpu (#38572)

Browse Source 
* 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>
This commit is contained in:
Yao Matrix
2025-06-06 15:29:51 +08:00
committed by GitHub
parent 31023b6909
commit 89542fb81c
23 changed files with 150 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/quantization/ggml/test_ggml.py
View File

@@ -20,7 +20,7 @@ from transformers import AddedToken, AutoModelForCausalLM, AutoModelForSeq2SeqLM
from transformers.testing_utils import (
require_gguf,
require_read_token,
require_torch_gpu,
require_torch_accelerator,
slow,
torch_device,
)
@@ -35,7 +35,7 @@ if is_gguf_available():
@require_gguf
@require_torch_gpu
@require_torch_accelerator
@slow
class GgufQuantizationTests(unittest.TestCase):
"""
@@ -107,7 +107,7 @@ class GgufQuantizationTests(unittest.TestCase):
@require_gguf
@require_torch_gpu
@require_torch_accelerator
@slow
class GgufIntegrationTests(unittest.TestCase):
"""
@@ -263,7 +263,7 @@ class GgufIntegrationTests(unittest.TestCase):
@require_gguf
@require_torch_gpu
@require_torch_accelerator
@slow
class GgufModelTests(unittest.TestCase):
mistral_model_id = "TheBloke/Mistral-7B-Instruct-v0.2-GGUF"