fix metadata feature

inisis · Nov 12, 2024 · 13318d6 · 13318d6
1 parent 8c82f56
commit 13318d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxslim/third_party/onnx_graphsurgeon/exporters/onnx_exporter.py b/onnxslim/third_party/onnx_graphsurgeon/exporters/onnx_exporter.py
@@ -367,7 +367,8 @@ def export_onnx(graph: Graph, do_type_check=True, **kwargs) -> "onnx.ModelProto"
         kwargs["ir_version"] = graph.ir_version
 
     model = onnx.helper.make_model(onnx_graph, **kwargs)
-    model.metadata_props.extend(graph.metadata_props)
+    if graph.metadata_props is not None:
+        model.metadata_props.extend(graph.metadata_props)
     model.producer_name = graph.producer_name
     model.producer_version = graph.producer_version
     return model