#!/usr/bin/env bash

set -euo pipefail

# Download the source model using:
#   hf download openai/whisper-medium \
#       --local-dir models--openai--whisper-medium

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
MODEL_PATH="models--openai--whisper-medium"
OUTPUT_PATH="whisper-medium-a16w8"
NUM_PROCESSES="${NUM_PROCESSES:-16}"

TMPDIR="$(mktemp -d /tmp/sima.XXXXXX)"
export TMPDIR

cleanup() {
    rm -rf -- "${TMPDIR}"
}
trap cleanup EXIT

python3 "${SCRIPT_DIR}/gen_models--openai--whisper.py" \
    --model_path "${MODEL_PATH}" \
    -o "${OUTPUT_PATH}" \
    --num_processes "${NUM_PROCESSES}" \
    --part all \
    --enable_log_probe \
    --resume \
    "$@"