From 682a8b625b410315c86a7504f91166d798fc5cfa Mon Sep 17 00:00:00 2001 From: aarushjain29 Date: Mon, 12 Jan 2026 20:18:34 +0000 Subject: [PATCH] opset change --- onnxruntime/python/tools/transformers/huggingface_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/huggingface_models.py b/onnxruntime/python/tools/transformers/huggingface_models.py index 4cd878a4656a7..8f4525037d66f 100644 --- a/onnxruntime/python/tools/transformers/huggingface_models.py +++ b/onnxruntime/python/tools/transformers/huggingface_models.py @@ -17,7 +17,7 @@ # Some models like GPT, T5, Bart etc has its own convert_to_onnx.py in models sub-directory, and they are excluded here. MODELS = { # BERT - "bert-base-cased": (["input_ids", "attention_mask", "token_type_ids"], 16, False, "bert"), + "bert-base-cased": (["input_ids", "attention_mask", "token_type_ids"], 18, False, "bert"), "bert-large-cased": (["input_ids", "attention_mask", "token_type_ids"], 16, False, "bert"), # Transformer-XL (Models uses Einsum, which need opset version 16 or later.) "transfo-xl-wt103": (["input_ids", "mems"], 16, False, "bert"),