Merge pull request #10 from abukhoy/pr-424

quic-rishinr · web-flow · commit 0c4604dd5007 · 2025-06-11T19:32:19.000+05:30
qnn tests fixed embedding test script
diff --git a/tests/transformers/models/test_embedding_models.py b/tests/transformers/models/test_embedding_models.py
@@ -142,7 +142,7 @@ def test_embed_model_pytorch_vs_onnx_vs_ai100_qnn(model_name):
     create_json(qnn_config_json_path, QnnConstants.QNN_SAMPLE_CONFIG)
 
     check_embed_pytorch_vs_ort_vs_ai100(
-        model_name=model_name, seq_len=32, n_layer=1, enable_qnn=True, qnn_config=qnn_config_json_path
+        model_name=model_name["model_name"], seq_len=32, n_layer=1, enable_qnn=True, qnn_config=qnn_config_json_path
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,7 @@ def test_embed_model_pytorch_vs_onnx_vs_ai100_qnn(model_name):`
`142`	`142`	`create_json(qnn_config_json_path, QnnConstants.QNN_SAMPLE_CONFIG)`
`143`	`143`
`144`	`144`	`check_embed_pytorch_vs_ort_vs_ai100(`
`145`		`- model_name=model_name, seq_len=32, n_layer=1, enable_qnn=True, qnn_config=qnn_config_json_path`
	`145`	`+ model_name=model_name["model_name"], seq_len=32, n_layer=1, enable_qnn=True, qnn_config=qnn_config_json_path`
`146`	`146`	`)`
`147`	`147`
`148`	`148`