Give library name when inferring task during export (#1115)

huggingface · Jan 16, 2025 · fe10aaa · fe10aaa
1 parent 7b4044d
commit fe10aaa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/exporters/openvino/__main__.py b/optimum/exporters/openvino/__main__.py
@@ -86,6 +86,7 @@ def infer_task(
                     revision=revision,
                     cache_dir=cache_dir,
                     token=token,
+                    library_name=library_name,
                 )
             except KeyError as e:
                 raise KeyError(