Skip to content

Commit 709560f

Browse files
authored
Qualcomm AI Engine Direct - Fix mobilebert finetune script (#9927)
Summary: - As title, it's broken in #9643
1 parent 491096e commit 709560f

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

examples/qualcomm/scripts/mobilebert_fine_tune.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
make_output_dir,
2424
make_quantizer,
2525
parse_skip_delegation_node,
26-
QnnPartitioner,
2726
setup_common_args_and_variables,
2827
SimpleADB,
2928
)
@@ -273,19 +272,15 @@ def calibrator(gm):
273272

274273
quantizer = make_quantizer(quant_dtype=quant_dtype)
275274
backend_options = generate_htp_compiler_spec(quant_dtype is not None)
276-
partitioner = QnnPartitioner(
277-
generate_qnn_executorch_compiler_spec(
278-
soc_model=getattr(QcomChipset, args.model),
279-
backend_options=backend_options,
280-
),
281-
skip_node_id_set=skip_node_id_set,
282-
skip_node_op_set=skip_node_op_set,
275+
compiler_specs = generate_qnn_executorch_compiler_spec(
276+
soc_model=getattr(QcomChipset, args.model),
277+
backend_options=backend_options,
283278
)
284279
# skip embedding layer cause it's quantization sensitive
285280
graph_module, _ = skip_annotation(
286281
nn_module=model,
287282
quantizer=quantizer,
288-
partitioner=partitioner,
283+
compiler_specs=compiler_specs,
289284
sample_input=inputs[0],
290285
calibration_cb=calibrator,
291286
fp_node_op_set={torch.ops.aten.embedding.default},

0 commit comments

Comments
 (0)