Skip to content

Commit 3d5d42f

Browse files
committed
save model first
1 parent c4e6215 commit 3d5d42f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

olive/passes/onnx/common.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,15 +213,16 @@ def ir_model_to_olive_model(
213213
if not isinstance(external_data_config, dict):
214214
external_data_config = external_data_config.dict()
215215
external_data_name = external_data_config.get("external_data_name")
216+
217+
ir.save(model, output_model_path, external_data=external_data_name)
218+
216219
if external_data_name:
217220
model_path = LocalFolder({"path": Path(output_model_path).parent})
218221
onnx_file_name = Path(output_model_path).name
219222
else:
220223
model_path = LocalFile({"path": output_model_path})
221224
onnx_file_name = None
222225

223-
ir.save(model, output_model_path, external_data=external_data_name)
224-
225226
return ONNXModelHandler(model_path=model_path, onnx_file_name=onnx_file_name)
226227

227228

0 commit comments

Comments
 (0)