#!/bin/bash
cd "$(dirname "$0")/.."
source env.sh
export CUDA_VISIBLE_DEVICES=$((SLURM_LOCALID*2))
exec .venv/bin/python -m viper.precompute \
  --clips data/annotated.jsonl --out_dir data/cache \
  --shard $SLURM_LOCALID --num_shards $SLURM_NTASKS