Skip to content

Commit 1b5acd1

Browse files
committed
Merge branch 'ea/gemma3' of https://github.com/eaidova/optimum-intel into ea/gemma3
2 parents 5642bea + 73892cf commit 1b5acd1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

optimum/exporters/openvino/model_configs.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -1158,7 +1158,7 @@ def patch_model_for_export(
11581158
library_name="transformers",
11591159
)
11601160
class Gemma3TextOpenVINOConfig(Gemma2OpenVINOConfig):
1161-
MIN_TRANSFORMERS_VERSION = version.parse("4.49.0")
1161+
MIN_TRANSFORMERS_VERSION = version.parse("4.50.0")
11621162

11631163

11641164
class DeciDummyPastKeyValuesGenerator(DummyPastKeyValuesGenerator):
@@ -3029,8 +3029,8 @@ def patch_model_for_export(
30293029

30303030

30313031
@register_in_tasks_manager("gemma3", *["image-text-to-text"], library_name="transformers")
3032-
class Gemma3OpneVINOConfig(LlavaOpenVINOConfig):
3033-
MIN_TRANSFORMERS_VERSION = "4.49.0"
3032+
class Gemma3OpenVINOConfig(LlavaOpenVINOConfig):
3033+
MIN_TRANSFORMERS_VERSION = "4.50.0"
30343034

30353035
def patch_model_for_export(
30363036
self, model: Union["PreTrainedModel", "TFPreTrainedModel"], model_kwargs: Optional[Dict[str, Any]] = None
@@ -3062,6 +3062,6 @@ def with_behavior(
30623062
self.int_dtype,
30633063
self.float_dtype,
30643064
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"}},
30663066
)
30673067
return super().with_behavior(behavior)

0 commit comments

Comments
 (0)