From dedb852eaea3d3d899920f4017b9fd899eb2756b Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Thu, 28 Mar 2024 01:12:24 +0800
Subject: [PATCH] Fix IR version for merged ONNX decoders (#1780)

fix IR version for merged decoders
---
 optimum/onnx/graph_transformations.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/optimum/onnx/graph_transformations.py b/optimum/onnx/graph_transformations.py
index 1c30644e21..ca08ae1aef 100644
--- a/optimum/onnx/graph_transformations.py
+++ b/optimum/onnx/graph_transformations.py
@@ -313,7 +313,10 @@ def merge_decoders(
             opset_imports.append(opset_import)
             opset_domains.add(opset_import.domain)
 
-    merged_model = onnx.helper.make_model(merged_graph, producer_name=producer_name, opset_imports=opset_imports)
+    # TODO: update IR version in the future.
+    merged_model = onnx.helper.make_model_gen_version(
+        merged_graph, producer_name=producer_name, opset_imports=opset_imports, ir_version=9
+    )
 
     check_and_save_model(merged_model, save_path=save_path)