#!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "$0")"

# SINGLE GPU only!
# multi-GPU sharding breaks Vamba's Mamba-2 state.

export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

python run.py --frame-budgets 64 128 256 "$@"