| """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) | |