#!/bin/bash
#SBATCH --job-name=vis_hal_circuit        # Job name
#SBATCH --output=./hallucination/scripts/out.txt      # Output file
#SBATCH --error=./hallucination/scripts/error.txt       # Error file
#SBATCH --ntasks=1               # Number of tasks (processes)
#SBATCH --gpus=1                 # Number of GPUs per node
#SBATCH --nodes=1               # Số node yêu cầu
#SBATCH --cpus-per-task=20       # Số CPU cho mỗi task

cd ..
declare -a text_sae_types=("crossattn")
declare -a vis_sae_types=("post")

python run_circuit_with_plot.py \
    --model_name "Salesforce/blip-image-captioning-base" \
    --tok_name "Salesforce/blip-image-captioning-base" \
    --data_path "./COCO-Dataset/filtered_val/hallucinated/75719" \
    --target_position -5 \
    --batch_size 16 \
    --text_sae_types "${text_sae_types[@]}" \
    --vis_sae_types "${vis_sae_types[@]}" \
    --text_batch 1000 \
    --vis_batch 200 \
    --crop_ratio 1.0 \
    --inter True \
    --intra True \
    --gradient_mode "ig" \
    --edge_gradient_mode "gradient" \
    --num_nodes 50 \
    --scoring_mode "less" \
    --ig_steps 10 \
    --filter_seq_length 30 \
    --save_name "vis_file/sae_graph_75719_nonhal.html" \
    --dtype "bfloat16" \
    --device_id 0