mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-04-20 00:51:35 +00:00
edf1ee1ef8
* whisper : improve whisper-cli executable path detection in model download shell scripts If whisper-cli is found on the path, do not suggest invoking from build directory. This improves flexibility and usability for distribution and packaging scenarios. * whisper : enhance Windows model download batch script to have comparable functionality and behaviour as shell scripts * Download models to the current directory if the script is executed from the \bin\ directory (for future distribution scenarios where the script is in the \bin\ subdirectory of a Windows build) * Add model_path command line argument * If whisper-cli is found on the path, do not suggest invoking from build directory * whisper : resolve compiler warning by removing duplicate definition of NOMINMAX in whisper-cli code
110 lines
2.7 KiB
Batchfile
110 lines
2.7 KiB
Batchfile
@echo off
|
|
|
|
rem Save the original working directory
|
|
set "orig_dir=%CD%"
|
|
|
|
rem Get the script directory
|
|
set "script_dir=%~dp0"
|
|
|
|
rem Check if the script directory contains "\bin\" (case-insensitive)
|
|
echo %script_dir% | findstr /i "\\bin\\" >nul
|
|
if %ERRORLEVEL%==0 (
|
|
rem If script is in a \bin\ directory, use the original working directory as default download path
|
|
set "default_download_path=%orig_dir%"
|
|
) else (
|
|
rem Otherwise, use script directory
|
|
pushd %~dp0
|
|
set "default_download_path=%CD%"
|
|
popd
|
|
)
|
|
|
|
rem Set the root path to be the parent directory of the script
|
|
for %%d in (%~dp0..) do set "root_path=%%~fd"
|
|
|
|
rem Count number of arguments passed to script
|
|
set argc=0
|
|
for %%x in (%*) do set /A argc+=1
|
|
|
|
set models=tiny tiny-q5_1 tiny-q8_0 ^
|
|
tiny.en tiny.en-q5_1 tiny.en-q8_0 ^
|
|
base base-q5_1 base-q8_0 ^
|
|
base.en base.en-q5_1 base.en-q8_0 ^
|
|
small small-q5_1 small-q8_0 ^
|
|
small.en small.en-q5_1 small.en-q8_0 ^
|
|
medium medium-q5_0 medium-q8_0 ^
|
|
medium.en medium.en-q5_0 medium.en-q8_0 ^
|
|
large-v1 ^
|
|
large-v2 large-v2-q5_0 large-v2-q8_0 ^
|
|
large-v3 large-v3-q5_0 ^
|
|
large-v3-turbo large-v3-turbo-q5_0 large-v3-turbo-q8_0
|
|
|
|
rem If argc is not equal to 1 or 2, print usage information and exit
|
|
if %argc% NEQ 1 (
|
|
if %argc% NEQ 2 (
|
|
echo.
|
|
echo Usage: download-ggml-model.cmd model [models_path]
|
|
CALL :list_models
|
|
goto :eof
|
|
)
|
|
)
|
|
|
|
if %argc% EQU 2 (
|
|
set models_path=%2
|
|
) else (
|
|
set models_path=%default_download_path%
|
|
)
|
|
|
|
set model=%1
|
|
|
|
for %%b in (%models%) do (
|
|
if "%%b"=="%model%" (
|
|
CALL :download_model
|
|
goto :eof
|
|
)
|
|
)
|
|
|
|
echo Invalid model: %model%
|
|
CALL :list_models
|
|
goto :eof
|
|
|
|
:download_model
|
|
echo Downloading ggml model %model%...
|
|
|
|
if exist "%models_path%\\ggml-%model%.bin" (
|
|
echo Model %model% already exists. Skipping download.
|
|
goto :eof
|
|
)
|
|
|
|
PowerShell -NoProfile -ExecutionPolicy Bypass -Command "Start-BitsTransfer -Source https://huggingface.co/ggerganov/whisper.cpp/resolve/main/ggml-%model%.bin -Destination \"%models_path%\\ggml-%model%.bin\""
|
|
|
|
if %ERRORLEVEL% neq 0 (
|
|
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
|
|
)
|
|
|
|
rem Check if 'whisper-cli' is available in the system PATH
|
|
where whisper-cli >nul 2>&1
|
|
if %ERRORLEVEL%==0 (
|
|
rem If found, suggest 'whisper-cli' (relying on PATH resolution)
|
|
set "whisper_cmd=whisper-cli"
|
|
) else (
|
|
rem If not found, suggest the local build version
|
|
set "whisper_cmd=%root_path%\build\bin\Release\whisper-cli.exe"
|
|
)
|
|
|
|
echo Done! Model %model% saved in %models_path%\ggml-%model%.bin
|
|
echo You can now use it like this:
|
|
echo %whisper_cmd% -m %models_path%\ggml-%model%.bin -f samples\jfk.wav
|
|
|
|
goto :eof
|
|
|
|
:list_models
|
|
echo.
|
|
echo Available models:
|
|
(for %%a in (%models%) do (
|
|
echo %%a
|
|
))
|
|
echo.
|
|
exit /b
|