extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-04-02 12:08:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

tests : add "threads" to run-tests.sh

Browse Source
This commit is contained in:
Georgi Gerganov 2023-04-29 12:32:18 +03:00
parent 3eaeb030ff
commit bab97c83d0
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/run-tests.sh
View File

@ -13,7 +13,7 @@
#
# Usage:
#
# ./tests/run-tests.sh <model_name>
# ./tests/run-tests.sh <model_name> [threads]
#
cd `dirname $0`
@ -32,7 +32,7 @@ function list_models {
}
if [ $# -eq 0 ]; then
printf "Usage: $0 [model]\n\n"
printf "Usage: $0 [model] [threads]\n\n"
printf "No model specified. Aborting\n"
list_models
exit 1
@ -41,6 +41,11 @@ fi
model=$1
main="../main"
threads=""
if [ $# -eq 2 ]; then
threads="-t $2"
fi
if [ ! -f ../models/ggml-$model.bin ]; then
printf "Model $model not found. Aborting\n"
list_models
@ -105,7 +110,7 @@ function run_lang() {
fi
fi
$main -m ../models/ggml-$model.bin -f $fname_dst -l $lang -otxt 2> /dev/null
$main -m ../models/ggml-$model.bin $threads -f $fname_dst -l $lang -otxt 2> /dev/null
git diff --no-index --word-diff=color --word-diff-regex=. $lang-$i-ref.txt $fname_dst.txt