#!/bin/bash
#SBATCH --job-name="gpt_7b"
#SBATCH --nodes=20
#SBATCH --nodelist=compute-h100-ord-node-[132,159,169,291,308,323,340,402,404,431,445,526,539,555,561,565,576,580,634,759,809,838,928,997]
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=64           # Number of cores per tasks
#SBATCH --gres=gpu:8                 # Number of gpus
#SBATCH --output=/app/suno/slurm/logs/run_%x_%j.txt   # Set this dir where you want slurm outs to go
#SBATCH --error=/app/suno/slurm/logs/run_%x_%j_err.txt    # Set this dir where you want slurm outs to go

# setup the environment using the script we created before

srun -K1 run_7b_tr.sh