#!/bin/bash
set -euo pipefail

export CUDA_VISIBLE_DEVICES="0,1"
export TOKENIZERS_PARALLELISM="false"

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/aokvqa/aokvqa_v1p0_train_conversations-towerplus-72B.jsonl \
    --output_file ${DATA_DIR}/interim/aokvqa/aokvqa_v1p0_train_conversations-towerplus-72B-juged.json \
    --images_dir ${IMAGE_DIR} \
    --judge_config config/vlm-judge_gemma-3-27B.yaml

