Skip to content

Commit 8345410

Browse files
Improve formatting of Gemma3OnnxConfig
1 parent 03d4de3 commit 8345410

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
@@ -507,9 +507,7 @@ class GemmaOnnxConfig(LlamaOnnxConfig):
507507
MIN_TRANSFORMERS_VERSION = version.parse("4.38.0")
508508

509509

510-
@register_tasks_manager_onnx(
511-
"gemma3", *[*COMMON_TEXT_GENERATION_TASKS, "text-classification"]
512-
)
510+
@register_tasks_manager_onnx("gemma3", *[*COMMON_TEXT_GENERATION_TASKS, "text-classification"])
513511
class Gemma3OnnxConfig(LlamaOnnxConfig):
514512
DUMMY_INPUT_GENERATOR_CLASSES = (
515513
DummyTextInputGenerator,

0 commit comments

Comments
 (0)