diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index bd8fd1de15..b09ec20a8c 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -1205,7 +1205,6 @@ class TasksManager: ), "vit-mae": supported_tasks_mapping( "feature-extraction", - "masked-im", onnx="VitMAEOnnxConfig", ), "vit-msn": supported_tasks_mapping(