diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index b09ec20a8c..7cb5a31d2d 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -1210,7 +1210,6 @@ class TasksManager: "vit-msn": supported_tasks_mapping( "feature-extraction", "image-classification", - "masked-im", onnx="VitMSNOnnxConfig", ), "vits": supported_tasks_mapping(