#!/bin/bash
##SBATCH --partition=unkillable
#SBATCH --job-name=laion-array
#SBATCH --output=output/gpu-%A_%a.out
#SBATCH --error=output/error/gpu-%A_%a.err
#SBATCH --gres=gpu:l40s:1
#SBATCH --cpus-per-gpu=4
#SBATCH --mem=32G
#SBATCH --time=48:00:00
#SBATCH --array=0    # 4 GPUs, one task per GPU

module load cuda/12.6.0
module load python/3.10
source $HOME/deep/bin/activate

python vlm_baseline.py