2022-10-30 19:51:29 +00:00
|
|
|
@echo off
|
|
|
|
|
|
|
|
pushd %~dp0
|
|
|
|
set models_path=%CD%
|
2022-12-08 04:18:30 +00:00
|
|
|
for %%d in (%~dp0..) do set root_path=%%~fd
|
2022-10-30 19:51:29 +00:00
|
|
|
popd
|
|
|
|
|
|
|
|
set argc=0
|
|
|
|
for %%x in (%*) do set /A argc+=1
|
|
|
|
|
2022-12-06 16:48:57 +00:00
|
|
|
set models=tiny.en tiny base.en base small.en small medium.en medium large-v1 large
|
2022-10-30 19:51:29 +00:00
|
|
|
|
|
|
|
if %argc% neq 1 (
|
|
|
|
echo.
|
|
|
|
echo Usage: download-ggml-model.cmd model
|
|
|
|
CALL :list_models
|
|
|
|
goto :eof
|
|
|
|
)
|
|
|
|
|
|
|
|
set model=%1
|
|
|
|
|
2022-11-15 17:47:06 +00:00
|
|
|
for %%b in (%models%) do (
|
2022-10-30 19:51:29 +00:00
|
|
|
if "%%b"=="%model%" (
|
|
|
|
CALL :download_model
|
|
|
|
goto :eof
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
echo Invalid model: %model%
|
|
|
|
CALL :list_models
|
|
|
|
goto :eof
|
|
|
|
|
|
|
|
:download_model
|
|
|
|
echo Downloading ggml model %model%...
|
|
|
|
|
|
|
|
cd %models_path%
|
|
|
|
|
|
|
|
if exist "ggml-%model%.bin" (
|
|
|
|
echo Model %model% already exists. Skipping download.
|
|
|
|
goto :eof
|
|
|
|
)
|
|
|
|
|
2023-01-02 07:54:43 +00:00
|
|
|
PowerShell -NoProfile -ExecutionPolicy Bypass -Command "Invoke-WebRequest -Uri https://huggingface.co/datasets/ggerganov/whisper.cpp/resolve/main/ggml-%model%.bin -OutFile ggml-%model%.bin"
|
2022-10-30 19:51:29 +00:00
|
|
|
|
2022-11-15 17:47:06 +00:00
|
|
|
if %ERRORLEVEL% neq 0 (
|
2022-10-30 19:51:29 +00:00
|
|
|
echo Failed to download ggml model %model%
|
|
|
|
echo Please try again later or download the original Whisper model files and convert them yourself.
|
|
|
|
goto :eof
|
|
|
|
)
|
|
|
|
|
2022-12-08 04:18:30 +00:00
|
|
|
echo Done! Model %model% saved in %root_path%\models\ggml-%model%.bin
|
2022-10-30 19:51:29 +00:00
|
|
|
echo You can now use it like this:
|
2022-12-08 04:18:30 +00:00
|
|
|
echo main.exe -m %root_path%\models\ggml-%model%.bin -f %root_path%\samples\jfk.wav
|
2022-10-30 19:51:29 +00:00
|
|
|
|
|
|
|
goto :eof
|
|
|
|
|
|
|
|
:list_models
|
|
|
|
echo.
|
|
|
|
echo Available models:
|
|
|
|
(for %%a in (%models%) do (
|
|
|
|
echo %%a
|
|
|
|
))
|
|
|
|
echo.
|
|
|
|
exit /b
|