WIP|

Author: eaidova

optimum/exporters/openvino/__main__.py

@@ -13,6 +13,7 @@
 #  limitations under the License.
 
 import gc
+import importlib
 import logging
 import operator
 import warnings
@@ -192,7 +193,6 @@ def main_export(
     ```
     """
     from optimum.exporters.openvino.convert import export_from_model
-
     if use_auth_token is not None:
         warnings.warn(
             "The `use_auth_token` argument is deprecated and will be removed soon. Please use the `token` argument instead.",
@@ -214,6 +214,7 @@ def main_export(
             revision=revision,
             cache_dir=cache_dir,
             token=token,
+            library_name=library_name
         )
         if library_name == "sentence_transformers":
             logger.warning(
@@ -233,6 +234,9 @@ def main_export(
         library_name=library_name,
     )
 
+    logger.warn(task)
+    logger.warn(library_name)
+
     do_gptq_patching = False
     do_quant_patching = False
     custom_architecture = False
@@ -447,6 +451,7 @@ class StoreAttr(object):
             device=device,
             trust_remote_code=trust_remote_code,
             patch_16bit_model=patch_16bit,
+            library_name=library_name,
             **kwargs_shapes,
         )
 

optimum/exporters/openvino/convert.py

@@ -606,6 +606,7 @@ def export_from_model(
     device: str = "cpu",
     trust_remote_code: bool = False,
     patch_16bit_model: bool = False,
+    library_name: Optional[str] = None,
     **kwargs_shapes,
 ):
     model_kwargs = model_kwargs or {}
@@ -615,9 +616,9 @@ 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 = _infer_library_from_model_or_model_class(model, library_name=library_name)
     if library_name != "open_clip":
-        TasksManager.standardize_model_attributes(model)
+        TasksManager.standardize_model_attributes(model, library_name=library_name)
 
     if hasattr(model.config, "export_model_type"):
         model_type = model.config.export_model_type.replace("_", "-")
@@ -630,7 +631,7 @@ def export_from_model(
         task = TasksManager.map_from_synonym(task)
     else:
         try:
-            task = TasksManager._infer_task_from_model_or_model_class(model=model)
+            task = TasksManager._infer_task_from_model_or_model_class(model=model, library_name=library_name)
         except (ValueError, KeyError) as e:
             raise RuntimeError(
                 f"The model task could not be automatically inferred in `export_from_model`. Please provide the argument `task` with the relevant task from {', '.join(TasksManager.get_all_tasks())}. Detailed error: {e}"