File tree 2 files changed +9
-1
lines changed
tests/transformers/models
2 files changed +9
-1
lines changed Original file line number Diff line number Diff line change 46
46
"ibm-granite/granite-guardian-3.1-2b" ,
47
47
]
48
48
49
+ test_models_qnn = [
50
+ "mistralai/Mixtral-8x7B-Instruct-v0.1" ,
51
+ "meta-llama/Llama-3.2-1B" ,
52
+ "unsloth/gemma-2b" ,
53
+ "ibm-granite/granite-guardian-3.1-2b" ,
54
+ ]
55
+
49
56
spd_test_models = [
50
57
"TinyLlama/TinyLlama-1.1B-Chat-v1.0" ,
51
58
]
@@ -241,7 +248,7 @@ def test_causal_lm_pytorch_vs_kv_vs_ort_vs_ai100(model_name):
241
248
242
249
@pytest .mark .on_qaic
243
250
@pytest .mark .qnn
244
- @pytest .mark .parametrize ("model_name" , test_models )
251
+ @pytest .mark .parametrize ("model_name" , test_models_qnn )
245
252
def test_causal_lm_pytorch_vs_kv_vs_ort_vs_ai100_qnn (model_name ):
246
253
"""
247
254
QNN Compilation Test
Original file line number Diff line number Diff line change @@ -366,6 +366,7 @@ def test_seq2seq_pytorch_vs_kv_vs_ort_vs_ai100(model_name):
366
366
367
367
@pytest .mark .on_qaic
368
368
@pytest .mark .qnn
369
+ @pytest .mark .skip (reason = "Whisper is currently not supported on QNN" )
369
370
@pytest .mark .parametrize ("model_name" , test_models )
370
371
def test_seq2seq_pytorch_vs_kv_vs_ort_vs_ai100_qnn (model_name ):
371
372
"""
You can’t perform that action at this time.
0 commit comments