Function install_composite_snapshot

tools/model_manager.py:1322–1389

(
    package: ModelPackage,
    source: CompositeSnapshotSource,
    models_root: Path,
    overwrite: bool,
)

            "model.safetensors",
            "tokenizer.json",
            "tokenizer_config.json",
            "audiovae.pth",
            "audiovae.safetensors",
        ),
    ),
    ModelPackage(
        id="voxcpm2_audiovae",
        display_name="VoxCPM2 AudioVAE local conversion utility",
        target_directory="VoxCPM2",
        source=ConverterSource(
            kind="pytorch_to_safetensors",
            description="Convert a local PyTorch checkpoint into safetensors.",
        ),
        required_files=("audiovae.safetensors",),
        description="Utility only. Use voxcpm2 for the full framework-ready VoxCPM2 package.",
    ),
    ModelPackage(
        id="htdemucs",
        display_name="HTDemucs",
        target_directory="htdemucs",
        source=ConverterSource(
            kind="demucs_reference",
            description="Download official Demucs .th checkpoints, or use --source-dir, and assemble framework safetensors plus a manifest.",
        ),
        required_files=(
            "manifest.json",
            "955717e8/config.json",
            "955717e8/model.safetensors",
        ),
    ),
)

PACKAGE_BY_ID = {package.id: package for package in CATALOG}


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(
        description="Download or convert official model packages into a local models layout."
    )
    subparsers = parser.add_subparsers(dest="command", required=True)

    list_parser = subparsers.add_parser("list", help="List available model packages.")
    list_parser.add_argument("--json", action="store_true", help="Print machine-readable JSON.")

    info_parser = subparsers.add_parser("info", help="Show one package.")
    info_parser.add_argument("package_id")
    info_parser.add_argument("--json", action="store_true", help="Print machine-readable JSON.")

    install_parser = subparsers.add_parser("install", help="Install one package into a models root.")
    install_parser.add_argument("package_id")
    install_parser.add_argument("--models-root", default="models", help="Target models root.")
    install_parser.add_argument("--overwrite", action="store_true", help="Replace an existing installed directory.")
    install_parser.add_argument("--source-file", help="Local source checkpoint for utility packages.")
    install_parser.add_argument("--output-file", help="Optional output file for single-file utility packages.")
    install_parser.add_argument("--source-dir", help="Optional local source directory for packages that can use checkpoint folders.")
    install_parser.add_argument("--variant", help="Variant or model name for packages that support variants.")

    return parser.parse_args()