diff_cli.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
"""aquin diff weight | sae | residue"""
 
import sys
 
 
def _print_help() -> None:
    print("Compare base vs checkpoint after fine-tuning.")
    print("")
    print("Usage: aquin diff <weight|sae|residue> ...")
    print("")
    print("  weight  Per-layer ||dW||, rank/collapse signals, merge verdict, behavioral probes")
    print("          aquin diff weight --checkpoint <path> [--prompts <json|jsonl>] [--no-behavioral]")
    print("")
    print("  sae     Feature-level activation deltas on probes")
    print("          aquin diff sae --checkpoint <path> [--prompts <json|jsonl>] [--layer N]")
    print("")
    print("  residue Per-layer activation drift on probes (base vs checkpoint)")
    print("          aquin diff residue --checkpoint <path> [--prompts <json|jsonl>]")
    print("")
    print("Docs: https://aquin.app/docs/checkpoint-sae")
 
 
def cmd_diff(args: list[str]) -> None:
    if not args or args[0] in ("-h", "--help", "help"):
        _print_help()
        return
 
    sub = args[0]
    rest = args[1:]
 
    if sub == "weight":
        from aquin.weight_diff_cli import cmd_weight_diff
 
        cmd_weight_diff(rest)
        return
 
    if sub == "sae":
        from aquin.sae_cli import cmd_sae_diff
 
        cmd_sae_diff(rest)
        return
 
    if sub == "residue":
        from aquin.residual_drift_cli import cmd_residual_drift
 
        cmd_residual_drift(rest)
        return
 
    print(f"Unknown diff subcommand: {sub}")
    print("Run: aquin diff --help")
    sys.exit(1)