Skip to content

Commit e9f5bdd

Browse files
Add supported tasks for Gemma3
1 parent 2a269df commit e9f5bdd

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

optimum/exporters/tasks.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -653,6 +653,13 @@ class TasksManager:
653653
"text-generation-with-past",
654654
"text-classification",
655655
),
656+
"gemma3": supported_tasks_mapping(
657+
"feature-extraction",
658+
"feature-extraction-with-past",
659+
"text-generation",
660+
"text-generation-with-past",
661+
"text-classification",
662+
),
656663
"glpn": supported_tasks_mapping(
657664
"feature-extraction",
658665
"depth-estimation",

0 commit comments

Comments
 (0)