File size: 546 Bytes
e0db531 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | #!/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
|