#!/bin/bash
#SBATCH --account=reformo
#SBATCH --partition=booster
#SBATCH --job-name=whisper-release-verify
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=16
#SBATCH --gres=gpu:4
#SBATCH --time=00:20:00
#SBATCH --time-min=00:05:00
#SBATCH --output=/e/scratch/reformo/schuhmann1_moss/whisper_score_regression/humaneness_ears_release_20261007/verify-%j.out
#SBATCH --error=/e/scratch/reformo/schuhmann1_moss/whisper_score_regression/humaneness_ears_release_20261007/verify-%j.err
set -euo pipefail
source /e/scratch/reformo/schuhmann1_moss/code/fastgen.sh
export PYTHONNOUSERSITE=1 OMP_NUM_THREADS=2 OPENBLAS_NUM_THREADS=1
export HF_HUB_OFFLINE=1 TRANSFORMERS_OFFLINE=1 TOKENIZERS_PARALLELISM=false
code=/e/home/jusers/schuhmann1/jupiter/whisper_score_regression/gemini_release
export RELEASE_CODE="$code"
srun --ntasks=1 python -m torch.distributed.run --standalone --nproc_per_node=2 "$code/export_rank.py"