From fe10aaaf3a96f7fc3b9645f2ef8a5bf6d2fb643f Mon Sep 17 00:00:00 2001
From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
Date: Thu, 16 Jan 2025 17:16:10 +0100
Subject: [PATCH] Give library name when inferring task during export (#1115)

---
 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 110ed4851..4047ab64a 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(