diff --git a/docs/source/exporters/onnx/overview.mdx b/docs/source/exporters/onnx/overview.mdx index a434094046..b5129c23f2 100644 --- a/docs/source/exporters/onnx/overview.mdx +++ b/docs/source/exporters/onnx/overview.mdx @@ -66,7 +66,7 @@ Supported architectures from [🤗 Transformers](https://huggingface.co/docs/tra - M2-M100 - Marian - MarkupLM -- Maskformer +- MaskFormer - MBart - MGP-STR - Mistral diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index 5c3269e193..771c9de70f 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -986,7 +986,7 @@ class MobileNetV2OnnxConfig(MobileNetV1OnnxConfig): pass -class MaskformerOnnxConfig(ViTOnnxConfig): +class MaskFormerOnnxConfig(ViTOnnxConfig): # torch.onnx.errors.UnsupportedOperatorError: Exporting the operator 'aten::einsum' to ONNX opset version 11 is not supported. # Support for this operator was added in version 12, try exporting with this version. DEFAULT_ONNX_OPSET = 12 diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index ab48154dda..bd8fd1de15 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -833,7 +833,7 @@ class TasksManager: "maskformer": supported_tasks_mapping( "feature-extraction", "image-segmentation", - onnx="MaskformerOnnxConfig", + onnx="MaskFormerOnnxConfig", ), "mbart": supported_tasks_mapping( "feature-extraction",