Skip to content

Commit 82bb6cc

Browse files
Improve formatting of Gemma3OnnxConfig
1 parent 16e5e8f commit 82bb6cc

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

optimum/exporters/onnx/model_configs.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -499,9 +499,7 @@ class GemmaOnnxConfig(LlamaOnnxConfig):
499499
MIN_TRANSFORMERS_VERSION = version.parse("4.38.0")
500500

501501

502-
@register_tasks_manager_onnx(
503-
"gemma3", *[*COMMON_TEXT_GENERATION_TASKS, "text-classification"]
504-
)
502+
@register_tasks_manager_onnx("gemma3", *[*COMMON_TEXT_GENERATION_TASKS, "text-classification"])
505503
class Gemma3OnnxConfig(LlamaOnnxConfig):
506504
DUMMY_INPUT_GENERATOR_CLASSES = (
507505
DummyTextInputGenerator,

0 commit comments

Comments
 (0)