#!/bin/bash
# Qualitative comparison set; 2 GPUs per task (one per Wan2.2 expert).
cd "$(dirname "$0")/.."
source env.sh
CKPT=${VIPER_CKPT:-checkpoints_interactive/stage1/final.pt}
export CUDA_VISIBLE_DEVICES=$((SLURM_LOCALID*2)),$((SLURM_LOCALID*2+1))
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export PYTHONUNBUFFERED=1
exec .venv/bin/python -m viper.make_comparison \
  --ckpt "$CKPT" --pairs data/pairs_val.jsonl --out_dir results/comparison \
  --steps 50 --guide_scale 6.0 \
  --shard $SLURM_LOCALID --num_shards $SLURM_NTASKS