Add gemma to skip check trace models (#1306)

echarlaix · web-flow · commit f03b0fea68c8 · 2025-05-14T14:07:16.000+02:00
diff --git a/optimum/exporters/openvino/utils.py b/optimum/exporters/openvino/utils.py
@@ -373,6 +373,9 @@ def set_simplified_chat_template(ov_tokenizer_model, processor_chat_template=Non
     "llama4",
 )
 
+if is_transformers_version("<", "4.41"):
+    SKIP_CHECK_TRACE_MODELS += ("gemma",)
+
 
 def allow_skip_tracing_check(library_name, model_type):
     if is_openvino_version("<", "2025.0.0"):

Original file line number	Diff line number	Diff line change
`@@ -373,6 +373,9 @@ def set_simplified_chat_template(ov_tokenizer_model, processor_chat_template=Non`
`373`	`373`	`"llama4",`
`374`	`374`	`)`
`375`	`375`
	`376`	`+if is_transformers_version("<", "4.41"):`
	`377`	`+ SKIP_CHECK_TRACE_MODELS += ("gemma",)`
	`378`	`+`
`376`	`379`
`377`	`380`	`def allow_skip_tracing_check(library_name, model_type):`
`378`	`381`	`if is_openvino_version("<", "2025.0.0"):`