Skip to content

Commit db52e05

Browse files
committed
fixed CI
1 parent a455f0d commit db52e05

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

backends/qualcomm/tests/test_qnn_delegate.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3154,9 +3154,9 @@ def test_llama3_2_1b(self):
31543154
"llama3_2",
31553155
"--model_mode",
31563156
"hybrid",
3157-
"--prefill_seq_len",
3157+
"--prefill_ar_len",
31583158
"32",
3159-
"--kv_seq_len",
3159+
"--max_seq_len",
31603160
"512",
31613161
"--num_sharding",
31623162
"4",
@@ -3234,9 +3234,9 @@ def test_llama_stories_110m(self):
32343234
"stories110m",
32353235
"--model_mode",
32363236
"hybrid",
3237-
"--prefill_seq_len",
3237+
"--prefill_ar_len",
32383238
"32",
3239-
"--kv_seq_len",
3239+
"--max_seq_len",
32403240
"128",
32413241
]
32423242
if self.compile_only:

0 commit comments

Comments
 (0)