diff --git a/optimum/intel/openvino/modeling_base.py b/optimum/intel/openvino/modeling_base.py index 97ad432fa6..feaed662a5 100644 --- a/optimum/intel/openvino/modeling_base.py +++ b/optimum/intel/openvino/modeling_base.py @@ -227,7 +227,7 @@ def _cached_file( model_cache_path = model_path / file_name else: file_name = Path(file_name) - if file_name.suffix != "onnx": + if file_name.suffix != ".onnx": model_file_names = [file_name.with_suffix(".bin"), file_name] else: model_file_names = [file_name]