Add CI test for Whisper models (#239)

.github/scripts/test-offline-whisper.sh

@@ -0,0 +1,85 @@
+#!/usr/bin/env bash
+
+set -e
+
+log() {
+  # This function is from espnet
+  local fname=${BASH_SOURCE[1]##*/}
+  echo -e "$(date '+%Y-%m-%d %H:%M:%S') (${fname}:${BASH_LINENO[0]}:${FUNCNAME[1]}) $*"
+}
+
+echo "EXE is $EXE"
+echo "PATH: $PATH"
+
+which $EXE
+
+names=(
+tiny.en
+base.en
+# small.en
+# medium.en
+)
+
+for name in ${names[@]}; do
+  log "------------------------------------------------------------"
+  log "Run $name"
+  log "------------------------------------------------------------"
+
+  repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-whisper-$name
+  log "Start testing ${repo_url}"
+  repo=$(basename $repo_url)
+  log "Download pretrained model and test-data from $repo_url"
+
+  GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url
+  pushd $repo
+  git lfs pull --include "*.onnx"
+  git lfs pull --include "*.ort"
+  ls -lh *.{onnx,ort}
+  popd
+
+  log "test fp32 onnx"
+
+  time $EXE \
+    --tokens=$repo/${name}-tokens.txt \
+    --whisper-encoder=$repo/${name}-encoder.onnx \
+    --whisper-decoder=$repo/${name}-decoder.onnx \
+    --num-threads=2 \
+    $repo/test_wavs/0.wav \
+    $repo/test_wavs/1.wav \
+    $repo/test_wavs/8k.wav
+
+  log "test int8 onnx"
+
+  time $EXE \
+    --tokens=$repo/${name}-tokens.txt \
+    --whisper-encoder=$repo/${name}-encoder.int8.onnx \
+    --whisper-decoder=$repo/${name}-decoder.int8.onnx \
+    --num-threads=2 \
+    $repo/test_wavs/0.wav \
+    $repo/test_wavs/1.wav \
+    $repo/test_wavs/8k.wav
+
+  log "test fp32 ort"
+
+  time $EXE \
+    --tokens=$repo/${name}-tokens.txt \
+    --whisper-encoder=$repo/${name}-encoder.ort \
+    --whisper-decoder=$repo/${name}-decoder.ort \
+    --num-threads=2 \
+    $repo/test_wavs/0.wav \
+    $repo/test_wavs/1.wav \
+    $repo/test_wavs/8k.wav
+
+  log "test int8 ort"
+
+  time $EXE \
+    --tokens=$repo/${name}-tokens.txt \
+    --whisper-encoder=$repo/${name}-encoder.int8.ort \
+    --whisper-decoder=$repo/${name}-decoder.int8.ort \
+    --num-threads=2 \
+    $repo/test_wavs/0.wav \
+    $repo/test_wavs/1.wav \
+    $repo/test_wavs/8k.wav
+
+  rm -rf $repo
+done

.github/workflows/linux.yaml

@@ -84,6 +84,14 @@ jobs:
           file build/bin/sherpa-onnx
           readelf -d build/bin/sherpa-onnx
 
+      - name: Test offline Whisper
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin:$PATH
+          export EXE=sherpa-onnx-offline
+
+          .github/scripts/test-offline-whisper.sh
+
       - name: Test offline CTC
         shell: bash
         run: |

.github/workflows/macos.yaml

@@ -82,6 +82,14 @@ jobs:
           otool -L build/bin/sherpa-onnx
           otool -l build/bin/sherpa-onnx
 
+      - name: Test offline Whisper
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin:$PATH
+          export EXE=sherpa-onnx-offline
+
+          .github/scripts/test-offline-whisper.sh
+
       - name: Test offline CTC
         shell: bash
         run: |

.github/workflows/windows-x64-cuda.yaml

@@ -74,6 +74,14 @@ jobs:
 
           ls -lh ./bin/Release/sherpa-onnx.exe
 
+      - name: Test offline Whisper for windows x64
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin/Release:$PATH
+          export EXE=sherpa-onnx-offline.exe
+
+          .github/scripts/test-offline-whisper.sh
+
       - name: Test offline CTC for windows x64
         shell: bash
         run: |

.github/workflows/windows-x64.yaml

@@ -75,6 +75,14 @@ jobs:
 
           ls -lh ./bin/Release/sherpa-onnx.exe
 
+      - name: Test offline Whisper for windows x64
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin/Release:$PATH
+          export EXE=sherpa-onnx-offline.exe
+
+          .github/scripts/test-offline-whisper.sh
+
       - name: Test offline CTC for windows x64
         shell: bash
         run: |

.github/workflows/windows-x86.yaml

@@ -73,6 +73,14 @@ jobs:
 
           ls -lh ./bin/Release/sherpa-onnx.exe
 
+      - name: Test offline Whisper for windows x86
+        shell: bash
+        run: |
+          export PATH=$PWD/build/bin/Release:$PATH
+          export EXE=sherpa-onnx-offline.exe
+
+          .github/scripts/test-offline-whisper.sh
+
       - name: Test offline CTC for windows x86
         shell: bash
         run: |