Add onnx export function for pix2struct model (#1815)

Add onnx export for pix2struct
huggingface · Apr 15, 2024 · 0b52e3a · 0b52e3a
1 parent 4936662
commit 0b52e3a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/onnxruntime/utils.py b/optimum/onnxruntime/utils.py
@@ -132,6 +132,7 @@ class ORTConfigManager:
         "vit": "vit",
         "whisper": "bart",
         "xlm-roberta": "bert",
+        "pix2struct": "vit",
     }
 
     @classmethod