diffusion_baseline / scripts /verify_conditional_prompts.py
jyliuAI's picture
Add files using upload-large-folder tool
840c3a4 verified
Raw History Blame Contribute Delete
1.21 kB
#!/usr/bin/env python3
"""Verify deterministic C64 conditional prompt artifacts before GPU work."""
from __future__ import annotations
import argparse
from pathlib import Path
import sys
def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
parser = argparse.ArgumentParser()
parser.add_argument("--root", type=Path, default=Path(__file__).resolve().parents[1])
parser.add_argument("--config", type=Path)
parser.add_argument("--dataset", choices=("all", "lm1b", "owt"), default="all")
return parser.parse_args(argv)
def main(argv: list[str] | None = None) -> int:
args = parse_args(argv)
root = args.root.resolve()
sys.path.insert(0, str(root / "src"))
from dlb.conditional_prompts import load_protocol, verify_prompts
protocol = load_protocol(args.config or root / "configs" / "conditional.yaml")
datasets = ("lm1b", "owt") if args.dataset == "all" else (args.dataset,)
for dataset_id in datasets:
manifest = verify_prompts(root, dataset_id, protocol)
print(f"OK {dataset_id} prompts={manifest.prompt_count} sha256={manifest.prompt_file_sha256}")
return 0
if __name__ == "__main__":
raise SystemExit(main())