fix typo

huggingface · Dec 18, 2024 · 7f0cb92 · 7f0cb92
1 parent a47cd96
commit 7f0cb92
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -833,7 +833,7 @@ class TasksManager:
         "maskformer": supported_tasks_mapping(
             "feature-extraction",
             "image-segmentation",
-            onnx="MaskformerOnnxConfig",
+            onnx="MaskFormerOnnxConfig",
         ),
         "mbart": supported_tasks_mapping(
             "feature-extraction",
-Original file line number
+Diff line change
@@ Expand Up @@
     - M2-M100
     - Marian
     - MarkupLM
-    - Maskformer
+    - MaskFormer
     - MBart
     - MGP-STR
     - Mistral
@@ Expand Down @@