ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-01-30 08:03:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bench-wts.sh : rename script + add execute permission

Browse Source
This commit is contained in:
Georgi Gerganov 2023-03-06 21:02:24 +02:00
parent b597c5a779
commit bb6b54a03d
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
extra/qual-bench.sh → extra/bench-wts.sh Normal file → Executable file
View File

@ -1,6 +1,8 @@
# Benchmark word-level timestamps for different models
#
# This script takes two arguments # This script takes two arguments
# - an audio file # - an audio file
# - [optional] path to a font file # - [optional] path to a font file
# I'm using "/usr/share/fonts/truetype/freefont/FreeMono.ttf" on Ubuntu # I'm using "/usr/share/fonts/truetype/freefont/FreeMono.ttf" on Ubuntu
@ -10,8 +12,9 @@ if [ -z "$1" ]; then
fi fi
#TODO: Make this a command line parameter #TODO: Make this a command line parameter
models="base small large" #models="base small large"
#models="tiny.en tiny base.en base small.en small medium.en medium large-v1 large" #models="tiny.en tiny base.en base small.en small medium.en medium large-v1 large"
models="tiny.en base.en small.en medium.en large"
DURATION=$(ffprobe -i $1 -show_entries format=duration -v quiet -of csv="p=0") DURATION=$(ffprobe -i $1 -show_entries format=duration -v quiet -of csv="p=0")
DURATION=$(printf "%.2f" $DURATION) DURATION=$(printf "%.2f" $DURATION)
@ -19,7 +22,7 @@ echo "Input file duration: ${DURATION}s"
for model in $models; do for model in $models; do
echo "Running $model" echo "Running $model"
COMMAND="./main -m models/ggml-$model.bin -owts -f $1 -of $1.$model" COMMAND="./main -m models/ggml-$model.bin -owts -f $1 -of $1.$model"
if [ ! -z "$2" ]; then if [ ! -z "$2" ]; then
COMMAND="$COMMAND -fp $2" COMMAND="$COMMAND -fp $2"
@ -40,7 +43,7 @@ for model in $models; do
# If the file already exists, delete it # If the file already exists, delete it
if [ -f $1.mp4 ]; then if [ -f $1.mp4 ]; then
rm $1.$model.mp4 rm $1.mp4
fi fi
bash $1.$model.wts >/dev/null 2>&1 bash $1.$model.wts >/dev/null 2>&1
@ -64,4 +67,4 @@ COMMAND="$COMMAND vstack=inputs=${COUNT}[v]\" -map \"[v]\" -map 1:a $1.all.mp4 >
echo $COMMAND echo $COMMAND
# Run the command # Run the command
eval $COMMAND eval $COMMAND