Merge pull request openvinotoolkit#80 from wgzintel/guozhong/print_ov…

…_version_in_convert Add openvino version printing in convert.py
wenyi5608 · Dec 15, 2023 · 650d421 · 650d421
2 parents 2c00bb4 + c3a096c
commit 650d421
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/llm_bench/python/benchmark.py b/llm_bench/python/benchmark.py
@@ -469,7 +469,7 @@ def main():
         os.system('echo OPENVINO_TORCH_BACKEND_DEVICE=$OPENVINO_TORCH_BACKEND_DEVICE')
 
     if framework == 'ov':
-        log.info(f'model_path={model_path}, openvino runtime version:{get_version()}')
+        log.info(f'model_path={model_path}, openvino runtime version: {get_version()}')
         if model_args['config'].get('PREC_BF16') and model_args['config']['PREC_BF16'] is True:
             log.warning('[Warning] Param bf16/prec_bf16 only work for framework pt. It will be disabled.')
     if args.memory_consumption:

diff --git a/llm_bench/python/convert.py b/llm_bench/python/convert.py
@@ -15,7 +15,7 @@
 from diffusers import UNet2DConditionModel, AutoencoderTiny, LCMScheduler
 from nncf import compress_weights
 from openvino import Type, PartialShape, save_model, convert_model
-from openvino.runtime import Core
+from openvino.runtime import Core, get_version
 from optimum.exporters import TasksManager
 from optimum.exporters.tasks import make_backend_config_constructor_for_task
 from optimum.exporters.onnx.config import TextDecoderOnnxConfig
@@ -1512,6 +1512,7 @@ def main():
     )
 
     args = parser.parse_args()
+    log.info(f'openvino runtime version: {get_version()}')
     model_type = get_convert_model_type(args.model_id.lower())
     converter = converters[model_type]
     converter(args)