Add ov model path attribute (#906)

huggingface · Sep 20, 2024 · 052c2b2 · 052c2b2
1 parent 40701f3
commit 052c2b2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/intel/openvino/modeling_base.py b/optimum/intel/openvino/modeling_base.py
@@ -74,6 +74,7 @@ def __init__(
         **kwargs,
     ):
         self.config = config
+        self.name_or_path = getattr(config, "name_or_path", None)
         self.model_save_dir = model_save_dir
         self._device = device.upper()
         self.is_dynamic = dynamic_shapes