zigzag9939's picture
Make experiments reproducible; link paper
dc9e1b9 verified
Raw History Blame Contribute Delete
977 Bytes
"""Hugging Face revisions the released lenses were fitted against.
Every model load in this directory passes revision=revision(model_id), so a
re-run downloads exactly these checkpoints (and, for Ouro/Huginn, exactly
this trust_remote_code). Mirrors MODEL_REVISIONS.md at the repo root.
"""
REVISIONS = {
"Qwen/Qwen3.6-27B": "6a9e13bd6fc8f0983b9b99948120bc37f49c13e9",
"ByteDance/Ouro-2.6B": "1ed04250da1a9936042725d302e81c8fa2ab5abd",
"ByteDance/Ouro-2.6B-Thinking": "f1edd81e7ac41355db670500ceaf204e0f73af68",
"tomg-group-umd/huginn-0125": "bb6621b65e90b6a4b9b29ef88dc83866d450470c",
}
# Commit of https://github.com/anthropics/jacobian-lens the code was run with.
JLENS_COMMIT = "581d398613e5602a5af361e1c34d3a92ea82ba8e"
def revision(model_id):
if model_id not in REVISIONS:
raise KeyError(f"no pinned revision for {model_id!r}; the released "
f"lenses only match {sorted(REVISIONS)}")
return REVISIONS[model_id]