Add config for generating ONNX models for table-transformers

huggingface · Dec 13, 2022 · a0a68c8 · a0a68c8
1 parent 9e56aff
commit a0a68c8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/transformers/onnx/features.py b/src/transformers/onnx/features.py
@@ -501,6 +501,12 @@ class FeaturesManager:
         "swin": supported_features_mapping(
             "default", "image-classification", "masked-im", onnx_config_cls="models.swin.SwinOnnxConfig"
         ),
+        "table-transformer": supported_features_mapping(
+            "default",
+            "object-detection",
+            "image-segmentation",
+            onnx_config_cls="models.table_transformer.TableTransformerOnnxConfig",
+        ),
         "t5": supported_features_mapping(
             "default",
             "default-with-past",