File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -213,15 +213,16 @@ def ir_model_to_olive_model(
213
213
if not isinstance (external_data_config , dict ):
214
214
external_data_config = external_data_config .dict ()
215
215
external_data_name = external_data_config .get ("external_data_name" )
216
+
217
+ ir .save (model , output_model_path , external_data = external_data_name )
218
+
216
219
if external_data_name :
217
220
model_path = LocalFolder ({"path" : Path (output_model_path ).parent })
218
221
onnx_file_name = Path (output_model_path ).name
219
222
else :
220
223
model_path = LocalFile ({"path" : output_model_path })
221
224
onnx_file_name = None
222
225
223
- ir .save (model , output_model_path , external_data = external_data_name )
224
-
225
226
return ONNXModelHandler (model_path = model_path , onnx_file_name = onnx_file_name )
226
227
227
228
You can’t perform that action at this time.
0 commit comments