Fix openvino export model from ONNX (#554)

huggingface · Feb 7, 2024 · e40e627 · e40e627
1 parent 24a1e30
commit e40e627
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]