#!/bin/bash
set -uo pipefail
export PRIME_API_KEY="$(cat "$AGENTPTB_PRIME_KEY_FILE")"
export TMPDIR=/var/lib/agentptb-cache/a/tmp
export CUDA_VISIBLE_DEVICES=${SFT_GPUS:-0,2,3}
export WANDB_MODE=disabled
cd /root/work/a/prime-rl
exec .venv/bin/sft @ "$@"