Create merge_onnx_ext_data.py

merge_onnx_ext_data.py

@@ -0,0 +1,25 @@
+"""
+Note: can also use external_data=False during export I think
+"""
+
+import onnx
+
+# ONNX model blueprint path (the small .onnx file)
+INPUT_MODEL_PATH = "/content/lsnet_xl_artist-dynamo-opset20.onnx"
+
+OUTPUT_MODEL_PATH = "/content/lsnet_xl_artist-dynamo-opset20_merged.onnx"
+
+print(f"Loading model from: {INPUT_MODEL_PATH}")
+# will automatically load the associated .onnx.data file
+model = onnx.load(INPUT_MODEL_PATH)
+
+print(f"Saving single-file model to: {OUTPUT_MODEL_PATH}")
+# This function re-embeds the external data into the main .onnx file
+onnx.save_model(
+    model,
+    OUTPUT_MODEL_PATH,
+    save_as_external_data=False,
+    all_tensors_to_one_file=True
+)
+
+print("finished.")