From e4b94a481fce8b1b79a70c21addfedb987f5720b Mon Sep 17 00:00:00 2001
From: Gabe Goodhart <ghart@us.ibm.com>
Date: Tue, 8 Oct 2024 14:51:57 -0600
Subject: [PATCH] fix: Add granite to the list of models that require
 position_ids

Branch: OnnxGranite

Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
---
 optimum/exporters/onnx/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/optimum/exporters/onnx/utils.py b/optimum/exporters/onnx/utils.py
index 675566ba23e..8df5ee4258a 100644
--- a/optimum/exporters/onnx/utils.py
+++ b/optimum/exporters/onnx/utils.py
@@ -86,6 +86,7 @@
     "phi",
     "phi3",
     "qwen2",
+    "granite",
 }