Add onnx export function for pix2struct model (#1815)

naormatania · web-flow · commit 0b52e3a39f65 · 2024-04-15T17:01:17.000+08:00
Add onnx export for pix2struct
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
