rust-whisper/rust-whisper.d/models/distil.sh

32 lines
736 B
Bash

#! /bin/bash
set -ueo pipefail
cd "$(dirname "$(realpath "$BASH_SOURCE")")"
if git lfs 2>&1 | grep not.a.git; then
brew install git-lfs
fi
echo "https://github.com/ggerganov/whisper.cpp/pull/1424" >&2
if ! test -d ./whisper; then
git clone https://github.com/openai/whisper
fi
if ! test -d ./whisper.cpp; then
git clone https://github.com/ggerganov/whisper.cpp
fi
cd ./whisper.cpp/models
for name in distil-medium.en; do # distil-large-v2
if ! test -d ./$name; then
git clone https://huggingface.co/distil-whisper/$name
fi
cd ./$name
git lfs install
git lfs pull
git lfs install
cd ..
python3 ./convert-h5-to-ggml.py ./$name/ ../../whisper .
mv ./ggml-model.bin ../../ggml-$name.bin
done