15 lines
510 B
Bash
15 lines
510 B
Bash
#!/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 |