#!/usr/bin/env bash

set -euo pipefail
cd "$(dirname "$0")"

export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0,1}"

# Avoid fragmentation OOMs for long-context forward passes
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
python run.py --frame-budgets 0 64 128 256 "$@"