Skip to content

Commit 9ef6766

Browse files
authored
Fix loading models from cache (#820)
1 parent 7238e74 commit 9ef6766

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

optimum/intel/openvino/modeling_base.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,11 @@ def fix_op_names_duplicates(model: openvino.runtime.Model):
132132

133133
if isinstance(file_name, str):
134134
file_name = Path(file_name)
135-
model = core.read_model(file_name) if not file_name.suffix == ".onnx" else convert_model(file_name)
135+
model = (
136+
core.read_model(file_name.resolve(), file_name.with_suffix(".bin").resolve())
137+
if not file_name.suffix == ".onnx"
138+
else convert_model(file_name)
139+
)
136140
if file_name.suffix == ".onnx":
137141
model = fix_op_names_duplicates(model) # should be called during model conversion to IR
138142

0 commit comments

Comments
 (0)