Skip to content

Commit 74d2161

Browse files
authored
fix diffusers version info in IR (#978)
1 parent 46e7781 commit 74d2161

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

optimum/exporters/openvino/convert.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
get_diffusion_models_for_export,
4040
)
4141
from optimum.intel.utils.import_utils import (
42+
_diffusers_version,
4243
_nncf_version,
4344
_open_clip_version,
4445
_optimum_intel_version,
@@ -806,7 +807,7 @@ def _add_version_info_to_model(model: Model, library_name: Optional[str] = None)
806807

807808
model.set_rt_info(sentence_transformers.__version__, ["optimum", "sentence_transformers_version"])
808809
if library_name == "diffusers":
809-
model.set_rt_info(_optimum_version, ["optimum", "diffusers_version"])
810+
model.set_rt_info(_diffusers_version, ["optimum", "diffusers_version"])
810811
elif library_name == "timm":
811812
model.set_rt_info(_timm_version, ["optimum", "timm_version"])
812813
elif library_name == "open_clip":

0 commit comments

Comments
 (0)