Skip to content

Commit 2230f91

Browse files
committed
Lint
1 parent c8b4d27 commit 2230f91

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

examples/models/llama/export_llama_lib.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -594,19 +594,14 @@ def _export_llama(modelname, args) -> LLMEdgeManager: # noqa: C901
594594
pt2e_quant_params, quantizers, quant_dtype = get_quantizer_and_quant_params(args)
595595

596596
# export_to_edge
597-
builder_exported = (
598-
_prepare_for_llama_export(modelname, args)
599-
.export()
600-
)
597+
builder_exported = _prepare_for_llama_export(modelname, args).export()
601598

602599
if args.export_only:
603600
exit()
604601

605-
builder_exported_to_edge = (
606-
builder_exported
607-
.pt2e_quantize(quantizers)
608-
.export_to_edge()
609-
)
602+
builder_exported_to_edge = builder_exported.pt2e_quantize(
603+
quantizers
604+
).export_to_edge()
610605

611606
modelname = builder_exported_to_edge.modelname
612607

0 commit comments

Comments
 (0)