File size: 518 Bytes
7bf8323 | 1 2 3 4 5 6 7 8 9 10 11 12 | """Merge single-bucket packages into one multifunction package (weights stored once; macOS 15+)."""
import sys
import coremltools as ct
out, specs = sys.argv[1], sys.argv[2:] # name=path ...
desc = ct.utils.MultiFunctionDescriptor()
for s in specs:
name, path = s.split("=")
desc.add_function(path, src_function_name="main", target_function_name=name)
desc.default_function_name = specs[1].split("=")[0] if len(specs) > 1 else specs[0].split("=")[0]
ct.utils.save_multifunction(desc, out)
print("saved", out)
|