From 01929b2e8773bfe11791e12f6ebbf7ce5db3c8a9 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Wed, 18 Dec 2024 17:23:13 +0100 Subject: [PATCH] remove vit-msn masked-im task --- optimum/exporters/tasks.py | 1 - 1 file changed, 1 deletion(-) 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(