liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
"""Audit byte-exact probes against several open tokenizer families."""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from transformers import AutoTokenizer
|
|
|
|
ROOT = Path(__file__).resolve().parent
|
|
MODELS = [
|
|
"Qwen/Qwen3-8B",
|
|
"Qwen/Qwen2.5-0.5B-Instruct",
|
|
"mistralai/Mistral-7B-v0.1",
|
|
]
|
|
|
|
|
|
def rows(path: Path):
|
|
return [json.loads(line) for line in path.read_text(encoding="utf-8").splitlines() if line.strip()]
|
|
|
|
|
|
def main():
|
|
probes = rows(ROOT / "data" / "eval.jsonl") + rows(ROOT / "data" / "boundary.jsonl")
|
|
report = {"probe_count": len(probes), "tokenizers": {}}
|
|
for model_id in MODELS:
|
|
tok = AutoTokenizer.from_pretrained(model_id, use_fast=True)
|
|
records = []
|
|
for row in probes:
|
|
ids = tok.encode(row["source"], add_special_tokens=False)
|
|
decoded = tok.decode(ids, skip_special_tokens=False)
|
|
records.append({"id": row["id"], "roundtrip": int(decoded == row["source"]), "tokens": len(ids)})
|
|
report["tokenizers"][model_id] = {
|
|
"vocab_size": len(tok),
|
|
"roundtrip_rate": sum(r["roundtrip"] for r in records) / len(records),
|
|
"mean_tokens": sum(r["tokens"] for r in records) / len(records),
|
|
"failures": [r["id"] for r in records if not r["roundtrip"]][:20],
|
|
}
|
|
out = ROOT / "validation" / "tokenizer_audit.json"
|
|
out.write_text(json.dumps(report, ensure_ascii=False, indent=2), encoding="utf-8")
|
|
print(json.dumps(report, ensure_ascii=False, indent=2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|