| |
| """Verify local sanitized LoRA files against weights_manifest.json.""" |
|
|
| from __future__ import annotations |
|
|
| import argparse |
| import hashlib |
| import json |
| from pathlib import Path |
|
|
|
|
| def sha256(path: Path) -> str: |
| digest = hashlib.sha256() |
| with path.open("rb") as handle: |
| for chunk in iter(lambda: handle.read(8 * 1024 * 1024), b""): |
| digest.update(chunk) |
| return digest.hexdigest() |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser(description="Verify HandEdit LoRA checkpoints.") |
| parser.add_argument( |
| "--root", |
| type=Path, |
| default=Path("."), |
| help="Repository root containing weights_manifest.json and checkpoints/.", |
| ) |
| args = parser.parse_args() |
|
|
| root = args.root.expanduser().resolve() |
| manifest = json.loads( |
| (root / "weights_manifest.json").read_text(encoding="utf-8") |
| ) |
| failures = [] |
| for adapter in manifest["adapters"].values(): |
| for item in adapter["files"]: |
| path = root / item["path"] |
| if not path.is_file(): |
| failures.append(f"missing: {item['path']}") |
| continue |
| if path.stat().st_size != item["size_bytes"]: |
| failures.append(f"size mismatch: {item['path']}") |
| continue |
| if sha256(path) != item["sha256"]: |
| failures.append(f"SHA-256 mismatch: {item['path']}") |
| continue |
| print(f"[OK] {item['path']}") |
|
|
| if failures: |
| raise SystemExit("\n".join(failures)) |
| print("[DONE] All sanitized LoRA files passed verification.") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
|
|
