fix: Add granite to the list of models that require position_ids

Branch: OnnxGranite Signed-off-by: Gabe Goodhart <[email protected]>
huggingface · Oct 10, 2024 · 29fc359 · 29fc359
1 parent 673c2c8
commit 29fc359
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/exporters/onnx/utils.py b/optimum/exporters/onnx/utils.py
@@ -86,6 +86,7 @@
     "phi",
     "phi3",
     "qwen2",
+    "granite",
 }