From 237b24e9f71e3fe82b432f115ced76295f09d1e2 Mon Sep 17 00:00:00 2001 From: eaidova Date: Wed, 30 Oct 2024 12:47:07 +0400 Subject: [PATCH] fix diffusers version info in IR --- optimum/exporters/openvino/convert.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py index 2c076827d2..df2885fd09 100644 --- a/optimum/exporters/openvino/convert.py +++ b/optimum/exporters/openvino/convert.py @@ -39,6 +39,7 @@ get_diffusion_models_for_export, ) from optimum.intel.utils.import_utils import ( + _diffusers_version, _nncf_version, _open_clip_version, _optimum_intel_version, @@ -806,7 +807,7 @@ def _add_version_info_to_model(model: Model, library_name: Optional[str] = None) model.set_rt_info(sentence_transformers.__version__, ["optimum", "sentence_transformers_version"]) if library_name == "diffusers": - model.set_rt_info(_optimum_version, ["optimum", "diffusers_version"]) + model.set_rt_info(_diffusers_version, ["optimum", "diffusers_version"]) elif library_name == "timm": model.set_rt_info(_timm_version, ["optimum", "timm_version"]) elif library_name == "open_clip":