batch_time_benchmark.sub

#!/bin/sh
#BSUB -q gpua100
#BSUB -J WhisperTranscript
#BSUB -n 12
#BSUB -gpu "num=1:mode=exclusive_process"
#BSUB -W 15:00
#BSUB -R "span[hosts=1]"
#BSUB -R "rusage[mem=8GB]"
#BSUB -R "select[gpu80gb]"
#BSUB -o runs/gpu_%J.out
# -- end of LSF options --

# Print information on gpu
nvidia-smi

#Load the necessary modules
module load cuda/11.7
module load python3/3.9.11
module load ffmpeg/4.4

# Activate the virtual environment -> link to your own before starting 
source /zhome/ad/4/137151/whsiperx/bin/activate

MODELS="large" #tiny base small medium 

for model in $MODELS
do
    python3 /zhome/ad/4/137151/Whisper-transcribe-tool/src/main.py model=$model benchmark=longform_time batch_size=128 whisper_version=whisperx
done