Add supported tasks for Gemma3

simondanielsson · simondanielsson · commit e9f5bddaacec · 2025-09-02T19:47:28.000+02:00
diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
@@ -653,6 +653,13 @@ class TasksManager:
             "text-generation-with-past",
             "text-classification",
         ),
+        "gemma3": supported_tasks_mapping(
+            "feature-extraction",
+            "feature-extraction-with-past",
+            "text-generation",
+            "text-generation-with-past",
+            "text-classification",
+        ),
         "glpn": supported_tasks_mapping(
             "feature-extraction",
             "depth-estimation",
