Skip to content

Commit 73bdb33

Browse files
authored
Update OnnxScriptFusion
1 parent b885198 commit 73bdb33

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

olive/passes/onnx/onnxscript_fusion.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,5 @@ def _run_for_config(
3838
# TODO(exporter team): Different fusions support different devices
3939
model_ir, function_stats = ort_fusions.optimize_for_ort(model_ir)
4040
logger.debug("Function stats: %s", function_stats)
41-
model_proto = ir.to_proto(model_ir)
4241
# save the model to the output path and return the model
43-
return model_proto_to_olive_model(model_proto, output_model_path, config)
42+
return model_ir_to_olive_model(model_ir, output_model_path)

0 commit comments

Comments
 (0)