Change openvino.runtime imports to openvino for 2025.0

tools/llm_bench/benchmark.py

@@ -6,7 +6,7 @@
 import argparse
 import logging as log
 import llm_bench_utils.model_utils
-from openvino.runtime import get_version
+from openvino import get_version
 import torch
 import traceback
 from llm_bench_utils.memory_profile import MemConsumption

tools/llm_bench/convert.py

@@ -20,7 +20,7 @@
 from diffusers import UNet2DConditionModel, AutoencoderTiny, LCMScheduler
 from nncf.torch.model_creation import is_wrapped_model
 from openvino import Type as OVType, PartialShape, save_model, convert_model
-from openvino.runtime import Core, get_version
+from openvino import Core, get_version
 from optimum.exporters import TasksManager
 from optimum.utils import DEFAULT_DUMMY_SHAPES
 from optimum.intel.openvino.configuration import OVConfig

tools/llm_bench/llm_bench_utils/ov_model_classes.py

@@ -16,7 +16,7 @@
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
 from optimum.intel.openvino import OVModelForCausalLM
 from optimum.intel.openvino.utils import ONNX_WEIGHTS_NAME, OV_XML_FILE_NAME
-from openvino.runtime import Model, Core, Tensor, Type
+from openvino import Model, Core, Tensor, Type
 from transformers import PretrainedConfig
 from transformers.modeling_outputs import CausalLMOutputWithPast, ModelOutput
 from transformers import GenerationConfig, StoppingCriteriaList

tools/llm_bench/llm_bench_utils/ov_utils.py

@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 from pathlib import Path
 from transformers import AutoConfig, AutoProcessor, AutoTokenizer
-from openvino.runtime import Core
+from openvino import Core
 import openvino as ov
 import logging as log
 import torch
@@ -17,7 +17,10 @@
     DEFAULT_MODEL_CLASSES,
     IMAGE_GEN_CLS
 )
-import openvino.runtime.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