enable qwen2 test

Signed-off-by: jiqing-feng <[email protected]>
huggingface · Jan 14, 2025 · 06798e2 · 06798e2
1 parent d3fbd65
commit 06798e2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py
@@ -233,8 +233,9 @@ class IPEXModelForCausalLMTest(unittest.TestCase):
         "distilgpt2",
         "mpt",
         "opt",
+        "qwen2",
     )
-    IPEX_PATCHED_SUPPORTED_ARCHITECTURES = ("llama2", "falcon", "gpt2")
+    IPEX_PATCHED_SUPPORTED_ARCHITECTURES = ("llama2", "falcon", "gpt2", "qwen2")
     GENERATION_LENGTH = 100
     SPEEDUP_CACHE = 1.0
 

diff --git a/tests/ipex/test_pipelines.py b/tests/ipex/test_pipelines.py
@@ -66,6 +66,7 @@ class PipelinesIntegrationTest(unittest.TestCase):
         "mistral",
         "mpt",
         "opt",
+        "qwen2",
     )
     QUESTION_ANSWERING_SUPPORTED_ARCHITECTURES = (
         "bert",

diff --git a/tests/ipex/utils_tests.py b/tests/ipex/utils_tests.py
@@ -64,4 +64,5 @@
     "patched_falcon": "Intel/tiny-random-falcon_ipex_model",
     "patched_gpt2": "Intel/tiny-random-gpt2_ipex_model",
     "patched_llama2": "Intel/tiny-random-llama2_ipex_model",
+    "qwen2": "Jiqing/tiny-random-Qwen2",
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -66,6 +66,7 @@ class PipelinesIntegrationTest(unittest.TestCase): @@
             "mistral",
             "mpt",
             "opt",
+            "qwen2",
         )
         QUESTION_ANSWERING_SUPPORTED_ARCHITECTURES = (
             "bert",
@@ Expand Down @@