wassname/jsteer
1
0
Fork 0
mirror of https://github.com/wassname/jsteer.git synced 2026-09-09 11:25:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

demo: fix demo_steer return annotation list->dict (stale after {summary,detail} change)

Browse Source 
Co-Authored-By: Claudypoo <288921227+claudypoo@users.noreply.github.com>
This commit is contained in:
wassname
2026-07-12 10:07:07 +08:00
co-authored by Claudypoo
parent caabfd3f12
commit da680901eb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jsteer/demo.py
+1 -1
View File
@@ -368,7 +368,7 @@ def show_steer(jac: Jacobian, model, tok, vec, user_msg: str, *,
@torch.no_grad()
def demo_steer(jac: Jacobian, model, tok, vecs: dict, user_msg: str, *,
rubric: str | None = None, readout: dict = DIGIT, budget: int = 6,
max_new_tokens: int = 256, seed: int = 0) -> list[dict]:
max_new_tokens: int = 256, seed: int = 0) -> dict:
"""THE steering demo, one call for every notebook. For each named vector: a clear
heading, a search for the strongest coherent steer each way, the raw generations at
[-C*, -C*/2, 0, +C*/2, +C*] (comparable qualitative outputs, same prompt), then ONE