Update tools/llm_bench/llm_bench_utils/ov_utils.py

Co-authored-by: Ekaterina Aidova <ekaterina.aidova@intel.com>
openvinotoolkit · Jan 17, 2025 · a567624 · a567624
1 parent 0113082
commit a567624
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tools/llm_bench/llm_bench_utils/ov_utils.py b/tools/llm_bench/llm_bench_utils/ov_utils.py
@@ -17,7 +17,10 @@
     DEFAULT_MODEL_CLASSES,
     IMAGE_GEN_CLS
 )
-import openvino.opset13 as opset
+try:
+    import openvino.opset13 as opset
+except ImportError:
+    import openvino.runtime.opset13 as opset
 from transformers import pipeline
 import openvino_genai as ov_genai
 import queue