#!/bin/bash

set -euo pipefail

export DATA_DIR="TODO: replace with your directory"
export CUDA_VISIBLE_DEVICES="0,1"

export TRANSLATION_MODEL="towerplus-9B"

python -m src.datasets.densefusion.make_translations \
    --input_path ${DATA_DIR}/raw/densefusion-1m/DenseFusion-1M.jsonl \
    --output_path ${DATA_DIR}/interim/densefusion-1m/DenseFusion-1M-${TRANSLATION_MODEL}.jsonl \
    --translator_config config/translation_${TRANSLATION_MODEL}.yaml \
    --comet_model "Unbabel/wmt23-cometkiwi-da-xl" \
