#!/bin/bash
set -euo pipefail

export CUDA_VISIBLE_DEVICES="0,1"

export DATA_DIR="TODO: replace with your directory"
export IMAGE_DIR="TODO: replace with your directory"

python -m src.datasets.allava.make_judgements \
    --input_file ${DATA_DIR}/interim/llava-instruct-150k/detail_23k_towerplus-9B.jsonl \
    --output_file ${DATA_DIR}/interim/llava-instruct-150k/detail_23k_towerplus-9B-juged.json \
    --images_dir ${IMAGE_DIR} \
    --judge_config config/vlm-judge_gemma-3-12B.yaml

python -m src.datasets.allava.make_judgements \
    --input_file ${DATA_DIR}/interim/llava-instruct-150k/conversation_58k_towerplus-9B.jsonl \
    --output_file ${DATA_DIR}/interim/llava-instruct-150k/conversation_58k_towerplus-9B-juged.json \
    --images_dir ${IMAGE_DIR} \
    --judge_config config/vlm-judge_gemma-3-27B.yaml

python -m src.datasets.allava.make_judgements \
    --input_file ${DATA_DIR}/interim/llava-instruct-150k/complex_reasoning_77k_towerplus-9B.jsonl \
    --output_file ${DATA_DIR}/interim/llava-instruct-150k/complex_reasoning_77k_towerplus-9B-juged.json \
    --images_dir ${IMAGE_DIR} \
    --judge_config config/vlm-judge_gemma-3-27B.yaml