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

Branch: OnnxGranite Signed-off-by: Gabe Goodhart <[email protected]>
huggingface · Oct 8, 2024 · e4b94a4 · e4b94a4
1 parent 88f4ace
commit e4b94a4
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",
 }