Update optimum/exporters/openvino/model_configs.py

eaidova · web-flow · commit 883534c57418 · 2025-03-14T10:03:33.000+04:00
diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py
@@ -3062,6 +3062,6 @@ def with_behavior(
                 self.int_dtype,
                 self.float_dtype,
                 model_patcher=Gemma3LMModelPatcher,
-                inputs_update={"token_type_ids": {0: "batch_size", 1: "past_sequence_length + 1"}},
+                inputs_update={"token_type_ids": {0: "batch_size", 1: "sequence_length"}},
             )
         return super().with_behavior(behavior)

Original file line number	Diff line number	Diff line change
`@@ -3062,6 +3062,6 @@ def with_behavior(`
`3062`	`3062`	`self.int_dtype,`
`3063`	`3063`	`self.float_dtype,`
`3064`	`3064`	`model_patcher=Gemma3LMModelPatcher,`
`3065`		`- inputs_update={"token_type_ids": {0: "batch_size", 1: "past_sequence_length + 1"}},`
	`3065`	`+ inputs_update={"token_type_ids": {0: "batch_size", 1: "sequence_length"}},`
`3066`	`3066`	`)`
`3067`	`3067`	`return super().with_behavior(behavior)`