Skip to content

Commit acc3390

Browse files
committed
move all conditions to allow_skip_check_trace
1 parent b0e5242 commit acc3390

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

optimum/exporters/openvino/convert.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@
4747
_transformers_version,
4848
compare_versions,
4949
is_openvino_tokenizers_version,
50-
is_openvino_version,
5150
is_tokenizers_version,
5251
is_transformers_version,
5352
)
@@ -438,9 +437,7 @@ def ts_patched_forward(*args, **kwargs):
438437
patcher.patched_forward = ts_patched_forward
439438

440439
ts_decoder_kwargs = {}
441-
if (library_name == "diffusers" or allow_skip_tracing_check(model)) and is_openvino_version(
442-
">=", "2025.0"
443-
):
440+
if allow_skip_tracing_check(library_name, model):
444441
ts_decoder_kwargs["trace_kwargs"] = {"check_trace": False}
445442

446443
with patcher:

optimum/exporters/openvino/utils.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from optimum.exporters import TasksManager
2727
from optimum.exporters.onnx.base import OnnxConfig
2828
from optimum.intel.utils import is_transformers_version
29-
from optimum.intel.utils.import_utils import is_safetensors_available
29+
from optimum.intel.utils.import_utils import is_openvino_version, is_safetensors_available
3030
from optimum.utils import is_diffusers_available
3131
from optimum.utils.save_utils import maybe_save_preprocessors
3232

@@ -350,7 +350,11 @@ def set_simplified_chat_template(ov_tokenizer_model, processor_chat_template=Non
350350
SKIP_CHECK_TRACE_MODELS = ("deepseek", "deepseek-v2", "deepseek-v3")
351351

352352

353-
def allow_skip_tracing_check(model):
353+
def allow_skip_tracing_check(library_name, model):
354+
if is_openvino_version("<", "2025.0.0"):
355+
return False
356+
if library_name == "diffusers":
357+
return True
354358
config = getattr(model, "config", {})
355359
model_type = getattr(config, "model_type", "").replace("_", "-")
356360
return model_type in SKIP_CHECK_TRACE_MODELS

0 commit comments

Comments
 (0)