#!/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