Skip to content

Commit ccbe97a

Browse files
committed
add patched qwen2 test
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent acfd0ce commit ccbe97a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/ipex/utils_tests.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@
5050
"mt5": "stas/mt5-tiny-random",
5151
"opt": "hf-internal-testing/tiny-random-OPTModel",
5252
"phi": "echarlaix/tiny-random-PhiForCausalLM",
53+
"qwen2": "Jiqing/tiny-random-Qwen2",
5354
"resnet": "hf-internal-testing/tiny-random-resnet",
5455
"roberta": "hf-internal-testing/tiny-random-roberta",
5556
"roformer": "hf-internal-testing/tiny-random-roformer",
@@ -64,5 +65,5 @@
6465
"patched_falcon": "Intel/tiny-random-falcon_ipex_model",
6566
"patched_gpt2": "Intel/tiny-random-gpt2_ipex_model",
6667
"patched_llama2": "Intel/tiny-random-llama2_ipex_model",
67-
"qwen2": "Jiqing/tiny-random-Qwen2",
68+
"patched_qwen2": "Jiqing/tiny-random-Qwen2_ipex_model",
6869
}

0 commit comments

Comments
 (0)