From 444bff7ddbc2215c2dc4f1414d40e6c76052613d Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Thu, 16 Jan 2025 15:50:14 +0100
Subject: [PATCH] give library name when inferring task during export

---
 optimum/exporters/openvino/__main__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/optimum/exporters/openvino/__main__.py b/optimum/exporters/openvino/__main__.py
index 110ed48515..4047ab64aa 100644
--- 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(