Skip to content

Commit 262a819

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

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

optimum/commands/optimum_cli.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,8 @@ def register_optimum_cli_subcommand(
175175

176176

177177
def main():
178+
print("HELLO from optimum")
179+
print()
178180
root = RootOptimumCLICommand("Optimum CLI tool", usage="optimum-cli")
179181
parser = root.parser
180182

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)