Merge branch 'master' into musicgen-onnx

huggingface · Mar 27, 2024 · b2e2f07 · b2e2f07
2 parents 082f043 + dedb852
commit b2e2f07
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)