Skip to content

Commit 3b2cffb

Browse files
authored
Allow skip trace check for sentence transformers models (#1332)
1 parent 8aa9390 commit 3b2cffb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/openvino/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ def set_simplified_chat_template(ov_tokenizer_model, processor_chat_template=Non
381381
def allow_skip_tracing_check(library_name, model_type):
382382
if is_openvino_version("<", "2025.0.0"):
383383
return False
384-
if library_name == "diffusers":
384+
if library_name in ["diffusers", "sentence_transformers"]:
385385
return True
386386
return model_type in SKIP_CHECK_TRACE_MODELS
387387

0 commit comments

Comments
 (0)