hujk/OGAAA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
HenryChou020514
2026-07-07 19:03:00 +08:00
commit 6edf7da2b7
158 changed files with 771425 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
Codes/2-2_head_identification/Ident_IH_04_visualize.sh Normal file
View File

@ -0,0 +1,15 @@
#!/usr/bin/env sh
set -eu
INPUT_PATH=${INPUT_PATH:-../Llama-3.1-8B-Instruct_head_scoring_raw.pkl}
MODEL_PATH=${MODEL_PATH:-}
PROMPT_INDEX=${PROMPT_INDEX:-2}
HEADS=${HEADS:-L0_H0}
OUTPUT_PATH="../head_heatmap.png"
python Ident_IH_04_visualize.py \
--input-dir ../2-2_head_identification/head_scoring/llama31-8b_injsq/heads_sorted \
--pickle-path ../2-2_head_identification/head_scoring/llama31-8b_injsq/head_scoring_raw_split_0.pkl \
--model-path ../../models/Llama-3.1-8B-Instruct \
--prompt-index 0