Update optimum/exporters/openvino/convert.py

eaidova · web-flow · commit 5f6c59714ea8 · 2025-02-04T12:42:02.000+04:00
diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py
@@ -616,7 +616,8 @@ def export_from_model(
             f"Compression of the weights to {ov_config.quantization_config} requires nncf, please install it with `pip install nncf`"
         )
 
-    library_name = _infer_library_from_model_or_model_class(model, library_name=library_name)
+   if library_name is None:
+        library_name = _infer_library_from_model_or_model_class(model)
     if library_name != "open_clip":
         TasksManager.standardize_model_attributes(model, library_name=library_name)
 

Original file line number	Diff line number	Diff line change
`@@ -616,7 +616,8 @@ def export_from_model(`
`616`	`616`	f"Compression of the weights to {ov_config.quantization_config} requires nncf, please install it with `pip install nncf`"
`617`	`617`	`)`
`618`	`618`
`619`		`- library_name = _infer_library_from_model_or_model_class(model, library_name=library_name)`
	`619`	`+ if library_name is None:`
	`620`	`+ library_name = _infer_library_from_model_or_model_class(model)`
`620`	`621`	`if library_name != "open_clip":`
`621`	`622`	`TasksManager.standardize_model_attributes(model, library_name=library_name)`
`622`	`623`